1,安装CUDA,下载,默认安装就行,注意安装目录,一般在这

"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA"

2,下载对应CUDA版本的CUDNN,下载完是个压缩包,解压

注意,这个要注册个账号,几分钟搞定

解压之后是这样的,把这3个目录下的所有文件,拷贝到CUDA安装目录的对应目录下,

比如bin目录下的文件,移动到CUDA安装目录下的bin目录下

3,安装 PyTorch GPU 版本,版本号不一样也没关系,比如我的是12.6,这里只支持12.4,那就用这个最新的12.4也行

安装的时候碰到报错

error: uninstall-no-record-file

× Cannot uninstall torch None
╰─> The package's contents are unknown: no RECORD file was found for torch.

和之前的有冲突,用pip也卸载不了,直接去torch的安装目录,手动删除

pip cache purge 清除缓存

再执行上面的安装命令就行了

4,试一下是否都安装正常,如下图,就是安装好了

import torchprint("CUDA available: ", torch.cuda.is_available())
print("CUDA version: ", torch.version.cuda)
print("cuDNN version: ", torch.backends.cudnn.version())
print("Device name: ", torch.cuda.get_device_name(0))