PyTorch中配置CUDA

第零部分：

首先查看CUDA版本

打开cmd，输入：nvidia-smi

可以看到CUDA的版本，后面安装的pytorch相关的库不高于这个版本的就行。

这里CUDA的版本是12.5，但是由于pytorch目前最高支持的cuda版本是12.1（在下文第二部分，图中最高只有12.1的），所以最高安装12.1的。

第一部分：

（好像可以直接安装第二部分，在pytorch中配置好支持cuda的包就行，不用再单独安装CUDA）

安装cuda和cudnn，可以参考：

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_pytorch cuda11.1百度网盘-CSDN博客

第二部分：

在官网Start Locally | PyTorch，找到下面的部分，按照你电脑的情况选择，比如package选项，如果安装了anoconda就选择conda。

在创建的环境的teminal中运行：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 

等待安装完成。

第三部分：

检验pytorch中是否有cuda环境：

运行以下代码：

import torch
print(torch.__version__)
print(torch.cuda.is_available())

输出： 

cuda环境配置完成。 

在运行目标检测yolo代码的时候，报错：Could not run 'torchvision::nms' with arguments from the 'CUDA' backend.

原来是安装的torchvision不支持cuda，检验如下：

import torch
import torchvision

print(torch.__version__)
print(torch.cuda.is_available())
print(torch.__version__)
print(torchvision.__version__)

可以看到torchvision安装的是cpu版本的。

torch和torchvision支持的cuda版本必须相同，不行的话就需要卸载重装，比如要重装torchvision，打开这个链接，找到对应版本的torchvision。

图中画出框表示：cuda的版本(cu121)，torchvision版本（0.18.1）和python版本(cp)。

下载好安装包，保存在项目文件夹中，然后在terminal中输入：

 pip install torchvision-0.18.1+cu121-cp312-cp312-win_amd64.whl

然后检验一下版本是否相同：

import torch
import torchvision

print(torch.__version__)
print(torch.cuda.is_available())
print(torchvision.__version__)

至此完成PyTorch中配置CUDA。

参考：

成功解决：NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend._notimplementederror: could not run 'torchvision::n-CSDN博客