使用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
选择自己的配置
一键安装
这里自用,不适用虚拟环境
出乎意外的居然没有毛病
开始测试
好像没啥毛病,能用。只是风扇一直转个不停了
import torch import torchvision print("TorchVision version:", torchvision.__version__) print("Is CUDA available:", torch.cuda.is_available()) print("CUDA version:", torch.version.cuda)
到此GPU环境搭建完成