1.手动安装cudnn:
https://blog.csdn.net/dhaduce/article/details/80379792
https://blog.csdn.net/dhaduce/article/details/80155121
以安装cudnn7.1为例
1. 官网下载对应的cudnn,一定要是arm版本的(aarch64)
官网地址cudnn下载地址TX2版-对应cuda-9.0:
http://developer.download.nvidia.com/compute/redist/cudnn/v7.0.5/cudnn-9.0-linux-x64-v7.tgz
http://developer.download.nvidia.com/compute/redist/cudnn/v7.1.1/cudnn-9.0-linux-x64-v7.1.tgz
这里注意了,只要去nvidia官网上去下载对应cuda版本Linux版本的cudnn即可。
2. 解压
cd /home/nvidia/Downloads
sudo tar xvf cudnn-9.0-linux-x64-v7.1.tgz
输出:
#cuda/include/cudnn.h
#cuda/NVIDIA_SLA_cuDNN_Support.txt
#cuda/lib64/libcudnn.so
#cuda/lib64/libcudnn.so.7
#cuda/lib64/libcudnn.so.7.1.1
#cuda/lib64/libcudnn_static.a
3. 复制过去
mkdir /home/nvidia/restful/cudnn7
sudo cp -r cuda /home/nvidia/restful/cudnn7/cuda
cd /home/nvidia/restful/cudnn7/cuda
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/* /usr/local/cuda/lib64/
cd /usr/local/cuda/lib64/
ls -a(查看libcudnn版本, 我的为7.1.1,以下对应位置都要是这个7.1.1;很多地方需要根据文件的版本然后操作,后面不再提示)
sudo chmod +r libcudnn.so.7.1.1
4. 删除&链接&生效
cd /usr/local/cuda/lib64
sudo rm -rf libcudnn.so libcudnn.so.7
sudo ln -s libcudnn.so.7.1.1 libcudnn.so.7
sudo ln -s libcudnn.so.7 libcudnn.so
sudo ldconfig
完成!!!
查看cuda版本:
nvcc -V
查询 cuda 版本号
cat /usr/local/cuda/version.txt
输出:
CUDA Version 8.0.61
显示cuda版本号为:8.0.61
查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
输出:
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 1
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include “driver_types.h”
cudnn 7.0.1