Windows下安装配置CUDA,过程并不复杂,也就是CUDA和cudnn这俩组件。
具体步骤如下:
- 打开Nvidia控制面板,帮助->系统信息->组件,找到3D设置里的NVCUDA64.DLL,看看产品名称那一栏对应的CUDA版本是多少,比如我的是 NVIDIA CUDA 11.7.101 driver,表示要用CUDA 11.7版本;
- 11.7的到这个地方下载CUDA并安装: https://developer.nvidia.com/cuda-11-7-0-download-archive ,其它版本的自己把链接里面的11-7改一下;
- 安装完成后,命令行中运行
nvcc --version
查看结果; - 再下载cudnn,地址 https://developer.nvidia.com/rdp/cudnn-download ;
- 将解压出来的cudnn的
bin
include
lib
目录下的文件分别复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
下对应的目录里,也可以直接把3个文件夹拷过去,合并即可; - 运行
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\extras\demo_suite\bandwidthTest.exe
查看结果;
我的电脑硬件:
- CPU:Xeon E5-2678 V3
- GPU:GTX 1050 Ti
- 内存:32G
最终在跑出来的PyTorch 1.13衣服分类成绩为
- cuda 65s
- cpu 68s
如果已经安装Linux子系统,也就是WSL2,在其中可以直接使用cuda,通过 nvidia-smi
可以查看到显卡信息。