可以指定pytorch版本下载到虚拟环境吗
时间: 2023-10-06 12:06:11 浏览: 78
python下不同深度学习环境的创建.doc
是的,可以通过指定版本号来下载特定版本的PyTorch。可以使用以下命令创建一个名为“myenv”的虚拟环境,并安装PyTorch 1.6.0:
```
conda create -n myenv python=3.6
conda activate myenv
conda install pytorch=1.6.0 torchvision=0.7.0 cudatoolkit=10.1 -c pytorch
```
请注意,这需要安装CUDA工具包10.1,如果您的GPU驱动程序不支持此版本,则需要安装适当的CUDA版本。
阅读全文