TX2手动安装cudnn

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
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值