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
#(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):复制过去
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
后面很多地方需要查看文件名称。学会使用 ls -a(查看当前目录下所有文件)
#(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
2.查看cuda版本:
nvcc -V
查询 cuda 版本号
命令行输入:cat /usr/local/cuda/version.txt
会输出如下信息:
CUDA Version 8.0.61
显示cuda版本号为:8.0.61
3.查看cudnn版本
查询 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