安装Pytorch CUDA版本

1.查看显卡Cuda版本

2.下载对应版本的torch、torchvision

下载网页

上图中，我的cuda版本为12.0.70，在下载网页中就查找cu12，但下载网页中没有，所以就找最相近的一版：
 

3.下载CUDA ToolKit

下载网页

4.卸载Python已安装的torch、torchvision

如果Python工程已经安装了cpu版的torch、torchvision，才执行这一步，否则跳过。

可通过pip命令来查看：

pip list

5.安装下载的torch、torchvision

通过以下命令来安装：

pip install torch的完整路径 torchvision的完整路径

6.安装CUDA ToolKit

双击运行下载的CUDA ToolKit应用程序即可，没有其他注意事项。

7.验证是否安装成功

同样使用第4步的pip命令行来查看已安装的库，如果torch后面带cu，就代表安装成功。