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

#(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

  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值