torch1.8.1支持cuda11.1吗
时间: 2023-09-25 22:13:19 浏览: 40
是的,torch 1.8.1 支持 CUDA 11.1。您可以在支持 CUDA 11.1 的计算机上安装 CUDA 11.1 和相应的 cuDNN 版本,然后使用 pip 安装 torch 1.8.1,即可在 CUDA 11.1 上运行 PyTorch。请注意,不同的操作系统和硬件可能需要不同的安装步骤和配置。
相关问题
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可以使用pytorch1.9.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,需要相应地修改这个参数。