使用pycharm 

1、CUDA的安装

官网：CUDA Toolkit Archive | NVIDIA Developer

选择对应的版本

选择对应的版本进行下载：

有3个多G

cuda的安装需要注意，如果没有安装vs，则需要选择自定义安装，在自定义的安装中取消

安装完成后的测试

其他的测试CUDA中有一个demo_suite文件夹中有一些demo测试

D:\program\Program Files\NVIDA GPU Computing Toolkit\CUDA\v12.3\extras\demo_suite

使用deviceQuery指令查询本机设备列表

2、下载cuDNN

官网：cuDNN 9.0.0 Downloads | NVIDIA Developer

下载完成后进行解压

拷贝文件

关于这两个东西是做什么的，后面在研究，先安装好

3、安装python GPU

官网：https://pytorch.org/

选择自己的配置

一键安装

这里自用，不适用虚拟环境

出乎意外的居然没有毛病

开始测试

好像没啥毛病，能用。只是风扇一直转个不停了

import torch
import torchvision
print("TorchVision version:", torchvision.__version__)
print("Is CUDA available:", torch.cuda.is_available())
print("CUDA version:", torch.version.cuda)

到此GPU环境搭建完成