pytorch1.8.1对应的CUDA是什么版本
时间: 2024-04-27 22:23:50 浏览: 15
PyTorch 1.8.1 对应的 CUDA 版本可以是 CUDA 10.2、CUDA 11.1、CUDA 11.2 或者 CUDA 11.3,具体取决于你的系统和硬件配置。你可以在 PyTorch 官网的安装页面中查看 PyTorch 版本和对应的 CUDA 版本。同时,还需要注意 PyTorch 版本和 CUDA 版本的兼容性,以免在使用过程中出现不兼容的情况。
相关问题
torch1.8.1对应cuda
根据引用和引用的信息,torch 1.8.1对应的cuda版本是10.2。在引用中提到的docker镜像中,pytorch 1.8.1对应的cuda版本是11.1。所以,根据不同的安装方式,torch 1.8.1可以与cuda 10.2或cuda 11.1配套使用。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* [Pytorch 1.8.1 with Cuda @ Windows 10](https://blog.csdn.net/blkf/article/details/115734972)[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^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"]
- *3* [Linux安装Pytorch1.8GPU(CUDA11.1)](https://blog.csdn.net/qq_37541097/article/details/114896215)[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^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"]
[ .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版本。