pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/cu111/torch_stable.html改成cuda11.4
时间: 2024-01-11 21:02:38 浏览: 391
torch_cluster-1.6.0-cp37-cp37m-linux_x86_64.whl.zip
如果你想安装的是CUDA 11.4版本的torch,可以使用以下命令:
```
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/cu114/torch_stable.html
```
这个命令会安装torch、torchvision和torchaudio,其中包含了cudnn模块所需的依赖。请注意,这个命令会安装CUDA 11.4版本的torch,如果你的CUDA版本不是11.4,需要在链接中选择相应的版本进行安装。
阅读全文