cuda11.1可以使用pytorch1.9.1么
时间: 2023-06-15 12:06:41 浏览: 157
1.10.0GPU版本pytorch环境,conda安装支持cuda10.2和11.3 pip安装支持cuda10.2和11.1
是的,可以使用。PyTorch 1.9.1官方支持CUDA 11.1。你可以通过pip安装PyTorch,然后使用CUDA 11.1来训练你的深度学习模型。以下是PyTorch官方文档中列出的支持的CUDA版本:
- PyTorch 1.9.1:CUDA 11.1
- PyTorch 1.9.0:CUDA 11.1
- PyTorch 1.8.1:CUDA 11.1
- PyTorch 1.8.0:CUDA 11.0
- PyTorch 1.7.1:CUDA 11.0
- PyTorch 1.7.0:CUDA 11.0
如果你计算机上已经安装了CUDA 11.1,可以通过以下命令安装PyTorch:
```
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
注意,这里使用了cu111,表示安装CUDA 11.1版本的PyTorch。如果你使用的是其他版本的CUDA,需要相应地修改这个参数。
阅读全文