第零部分：
首先查看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博客
