我想要安装cuda11.5对应的pytorch
时间: 2023-09-20 13:10:20 浏览: 1734
PyTorch官方文档v0.1.11_5
4星 · 用户满意度95%
可以按照以下步骤安装:
1. 安装 CUDA 11.5。可以从 NVIDIA 官网下载安装包并按照提示进行安装。
2. 安装 cuDNN 8.2。同样可以从 NVIDIA 官网下载安装包并按照提示进行安装。
3. 创建一个虚拟环境(可选)。
```
conda create -n pytorch_env python=3.8
conda activate pytorch_env
```
4. 安装 PyTorch。
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
```
这将安装 CUDA 11.1 对应的 PyTorch 版本。如果你想安装 CUDA 11.5 对应的版本,可以使用以下命令:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.5 -c pytorch -c nvidia
```
安装完成后,可以通过以下代码测试 PyTorch 是否成功安装:
```
import torch
if torch.cuda.is_available():
print('CUDA is available!')
else:
print('CUDA is not available.')
```
如果输出结果为 `CUDA is available!`,则说明 PyTorch 安装成功,并且可以使用 GPU 进行计算。
阅读全文