pip install torch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 -i https://mirrors.aliyun.com/pypi/simple/这样可以下载GPU版本的pytorch吗
时间: 2024-01-11 21:02:40 浏览: 160
安装pytorch-gpu版本(分享我的成功案例)
5星 · 资源好评率100%
不可以,因为目前最新的PyTorch版本是1.9.0,而且不支持Python 2.0.0版本。另外,这个命令中提供的镜像源是阿里云的镜像源,可以加速下载速度,但不影响版本的选择。如果想要下载GPU版本的PyTorch,应该使用以下命令:
```
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu102/torch_stable.html
```
其中,`cu102`表示CUDA的版本号。如果你的CUDA版本不是10.2,可以根据实际情况替换成相应的版本号。
阅读全文