linux cuda11.7对应的pytorch
时间: 2023-09-23 16:07:39 浏览: 308
机器学习,pytorch英伟达显卡驱动,cndann,适用于cuda12.1以上
根据官方文档,目前(2022年1月)最新的PyTorch版本是1.10,它支持CUDA 11.1、11.2、11.3、11.4和11.5,但不支持CUDA 11.7。因此,如果你需要使用PyTorch,建议安装CUDA 11.5或更早版本。在安装PyTorch时,可以指定CUDA版本,例如:
```bash
pip install torch==1.10.0+cu111 -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
其中,cu111表示CUDA 11.1,你可以将它替换为cu110、cu102等其他版本。
阅读全文