系统ubuntu18.04
机器上有两个nvidia2080ti,有人已经装了nvidia driver和cuda11.0,没看到dudnn。
于是就想把cudnn装上,在网上看了一下dudnn8.0与之匹配。
到nvidia官网上下载:
在这里下载需要注册,我以前用邮箱注册过,但记不住密码了。重新用微信注册了一个账号。
https://developer.nvidia.com/rdp/cudnn-download
找到:
Download cuDNN v8.0.4 (September 28th, 2020), for CUDA 11.0
分别点开:
cuDNN Runtime Library for Ubuntu18.04 x86_64 (Deb)
cuDNN Developer Library for Ubuntu18.04 x86_64 (Deb)
cuDNN Code Samples and User Guide for Ubuntu18.04 x86_64 (Deb)
下载了这三个文件。
在ubuntu下用sudo dpkg -i ....
安装。
最后,用:
cp -r /usr/src/cudnn_samples_v7/ $HOME
cd $HOME/cudnn_samples_v7/mnistCUDNN
./make clean && make
几条命令测试:
...
Resulting weights from Softmax:
0.0000000 0.0000000 0.0000000 1.0000000 0.0000000 0.0000714 0.0000000 0.0000000 0.0000000 0.0000000
Loading image data/five_28x28.pgm
Performing forward propagation ...
Resulting weights from Softmax:
0.0000000 0.0000008 0.0000000 0.0000002 0.0000000 1.0000000 0.0000154 0.0000000 0.0000012 0.0000006
Result of classification: 1 3 5
Test passed!
完美通过!