Jetson AGX Xavier配置cuDNN流程

Jetson AGX Xavier配置cuDNN流程

环境：Jetson AGX Xavier，CUDA 10.2，cuDNN 8.0.0

1 软件版本查询

CUDA版本查询：

nvcc -V

或者，cat /usr/local/cuda/version.txt

 

cuDNN版本查询：

cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

2 配置cuDNN

2.1 复制头文件与库文件

Xavier虽然安装了cuDNN，但没有将对应的头文件、库文件放到cuda目录。Xavier中cuDNN的头文件在：/usr/include，库文件位于：/usr/lib/aarch64-linux-gnu。将头文件与库文件复制到cuda目录下：

cd /usr/include && sudo cp cudnn.h /usr/local/cuda/include

cd /usr/lib/aarch64-linux-gnu && sudo cp libcudnn* /usr/local/cuda/lib64

2.2 修改文件权限

修改复制完的头文件与库文件的权限，所有用户都可读，可写，可执行：

sudo chmod 777 /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

2.3 重新链接

cd /usr/local/cuda/lib64

sudo ln -sf libcudnn.so.8.0.0 libcudnn.so.8

sudo ln -sf libcudnn_ops_train.so.8.0.0 libcudnn_ops_train.so.8

sudo ln -sf libcudnn_ops_infer.so.8.0.0 libcudnn_ops_infer.so.8

sudo ln -sf libcudnn_adv_infer.so.8.0.0 libcudnn_adv_infer.so.8

sudo ln -sf libcudnn_cnn_infer.so.8.0.0 libcudnn_cnn_infer.so.8

sudo ln -sf libcudnn_cnn_train.so.8.0.0 libcudnn_cnn_train.so.8

sudo ln -sf libcudnn_adv_train.so.8.0.0 libcudnn_adv_train.so.8

如果不重新链接，直接ldconfig，会出现“libcudnn.so.8 不是符号链接”，“libcudnn_ops_train.so.8 不是符号链接”，“libcudnn_ops_infer.so.8 不是符号链接”，“libcudnn_adv_infer.so.8 不是符号链接”，“libcudnn_cnn_infer.so.8 不是符号链接”，“libcudnn_cnn_train.so.8 不是符号链接”，“libcudnn_adv_train.so.8 不是符号链接”，如下图所示：

2.4 编译

sudo ldconfig

3 测试cuDNN

sudo cp -r /usr/src/cudnn_samples_v8/ ~/

cd ~/cudnn_samples_v8/mnistCUDNN

sudo chmod 777 ~/cudnn_samples_v8

sudo make clean && make

./mnistCUDNN

如果配置成功 测试完成后会显示：“Test passed!”，如下图所示：

 

4 参考文献

1. Nvidia官网cuDNN配置教程：https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#installdriver
2. 源码安装cuDNN v7.3.0 for CUDA 10.0，并测试cuDNN是否安装成功：https://blog.csdn.net/xingdou520/article/details/83902685
3. jetson xavier 配置 Tensorflow1.12.0（CUDA 10.0 + cuDNN 7.3.0 )：https://blog.csdn.net/xingdou520/article/details/84402577#t6
4. Linux ln命令：https://www.runoob.com/linux/linux-comm-ln.html
5. Linux ldconfig命令：https://ipcmen.com/ldconfig
6. Linux chmod命令用法：https://www.cnblogs.com/xiaoxiaoweng/p/10722044.html
7. Ubuntu: ldconfig(解决*.so不是符号连接)：https://blog.csdn.net/wishchin/article/details/91528315
8. 解决：mkdir: cannot create directory ‘test’: Permission denied：https://blog.csdn.net/weixin_42433970/article/details/102462961
9. linux下 cat grep快速查找内容：https://blog.csdn.net/yuan882696yan/article/details/81663579