如果cuda岁11.4应该安装哪个版本的pytorch
时间: 2023-09-26 20:07:00 浏览: 100
pytorch安装pytorch+gpu版本安装,pytorch+cuda10.1+cudnn7.6.5安装
如果你的系统安装了CUDA 11.4,则可以使用以下命令安装适用于Python 3.8的PyTorch:
```
pip install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
请注意,这需要使用适用于CUDA 11.1或更高版本的NVIDIA驱动程序和cuDNN版本。如果你的系统上安装的驱动程序版本与此不匹配,则可能需要更新驱动程序和cuDNN。
阅读全文