Nvidia现在要求安装CuDNN和NICC都得登录Nvidia Developer网站,国内巨慢。药丸。
¶CUDA
1 | sudo -i # 切换root权限 |
¶NICC
1 | apt install libnccl2=2.15.5-1+cuda11.8 libnccl-dev=2.15.5-1+cuda11.8 |
¶cuDNN
1 | dpkg -i cudnn-local-repo-ubuntu2204-8.6.0.163_1.0-1_amd64.deb |
¶测试cuDNN
mnistCUDNN测试需要FreeImage。
1 | apt install libfreeimage3 libfreeimage-dev |
将sample拷贝到有写权限的目录,并正确设置权限。
1 | cp /usr/src/cudnn_samples_v8/ . -R |
进入每个目录make
生成可执行文件,运行。如都可以运行,显示Test passed!
,则安装成功。