Windows下安装配置CUDA

Windows下安装配置CUDA,过程并不复杂,也就是CUDA和cudnn这俩组件。

具体步骤如下:

  1. 打开Nvidia控制面板,帮助->系统信息->组件,找到3D设置里的NVCUDA64.DLL,看看产品名称那一栏对应的CUDA版本是多少,比如我的是 NVIDIA CUDA 11.7.101 driver,表示要用CUDA 11.7版本;
  2. 11.7的到这个地方下载CUDA并安装: https://developer.nvidia.com/cuda-11-7-0-download-archive ,其它版本的自己把链接里面的11-7改一下;
  3. 安装完成后,命令行中运行nvcc --version查看结果;
  4. 再下载cudnn,地址 https://developer.nvidia.com/rdp/cudnn-download
  5. 将解压出来的cudnn的bin include lib目录下的文件分别复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA下对应的目录里,也可以直接把3个文件夹拷过去,合并即可;
  6. 运行C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\extras\demo_suite\bandwidthTest.exe查看结果;

我的电脑硬件:

  1. CPU:Xeon E5-2678 V3
  2. GPU:GTX 1050 Ti
  3. 内存:32G

最终在跑出来的PyTorch 1.13衣服分类成绩为

  1. cuda 65s
  2. cpu 68s

如果已经安装Linux子系统,也就是WSL2,在其中可以直接使用cuda,通过 nvidia-smi可以查看到显卡信息。

Leave a Comment

豫ICP备19001387号-1