cuda11.1对应pytorch版本
时间: 2023-09-30 17:02:22 浏览: 53
PyTorch版本与CUDA版本之间的对应关系并不固定,需要查看PyTorch的官方文档。
根据PyTorch的官方文档,在2021年的最新版本PyTorch 1.7.0中,支持CUDA 11.0和11.1。因此,可以使用PyTorch 1.7.0与CUDA 11.1配合使用。
不过,建议您确认您的计算机环境是否符合PyTorch 1.7.0的系统要求,以确保安装和使用的顺利。
相关问题
cuda11.1对应的pytorch版本
根据引用\[1\]和引用\[2\]的信息,可以得知cuda11.1对应的pytorch版本是1.8.2。因此,你可以使用以下命令安装对应版本的pytorch:
pip3 install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
这样就可以安装cuda11.1对应的pytorch版本了。
#### 引用[.reference_title]
- *1* [CUDA11.1对应pytorch安装](https://blog.csdn.net/qq_45919842/article/details/127631944)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
- *2* [pytorch+cuda11.1安装问题及解决(pycharm)](https://blog.csdn.net/qq_43605392/article/details/124108229)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
- *3* [配置pytorch(cuda11.1)](https://blog.csdn.net/weixin_58577392/article/details/124115119)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
CUDA11.1对应的pytorch版本
对应CUDA 11.1的PyTorch版本有两个:
1. PyTorch 1.8.1,支持CUDA 11.1和cuDNN 8.0及以上版本。
2. PyTorch 1.9.0rc1,支持CUDA 11.1和cuDNN 8.0及以上版本。
您可以根据自己的需要选择其中一个版本进行安装。建议先检查系统环境和CUDA版本,以便选择正确的PyTorch版本。
阅读全文