jetson tk1开发（2）-安装cuda和cudnn

来源：互联网发布：jqueryrotate.js教程编辑：程序博客网时间：2024/05/17 23:15

这一步为安装tensorflow前的预备工作，需要注意的是以下软件的安装版本号，不同版本之间经常会发现不兼容的问题。

首先安装CUDA，版本选择为6.5：

1、下载安装包：https://developer.nvidia.com/cuda-toolkit-65，下载下来为xxx.deb

2、解压缩：

sudo dpkg -i xxx.deb

3、更新软件源（有问题的可查看我下一章博客）

sudo apt-get update

4、安装toolkit:

sudo apt-get install cuda-toolkit-6-5

5、设置当前用户下可以访问GPU

sudo usermod -a -G video $USER

6、配置环境变量：

vim .bashrc

添加下面两行代码（具体地址参考自己安装的位置）：

export PATH=/usr/local/cuda/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda/lib:$LD_LIBRARY_PATH

退出进行保存：

:wq

source ~/.bashrc

7、验证是否安装成功

 nvcc -V

安装cuDNN，版本号v2（版本v1不兼容tensorflow）
1、建立目录：

mkdir /cuDNNR2

2、下载软件包
软件包名称：cudnn-6.5-linux-ARMv7-R2-rc1.tgz
下载地址：https://developer.nvidia.com/cuDNN

3、在软件包目录上写一个sh脚本,命名为installcuDNNR2.sh
脚本内容：

#!/bin/sh# Install cuDNN R2 on NVIDIA Jetson TK1# Register as a NVIDIA developer and download the cuDNN package# Package is named cudnn-6.5-linux-ARMv7-R2-rc1.tgz# This script and the cuDNN package should be in the same directory, usually ~/Downloads# This script places the library and include files for cudnn in the CUDA 6.5 directories# Make sure this is executing as rootif [ $(id -u) != 0 ]; then  echo "This script requires root permissions"  echo "$ sudo "$0""  exitfi# unzip the archivetar -zxvf cudnn-6.5-linux-ARMv7-R2-rc1.tgzcd cudnn-6.5-linux-ARMv7-R2-rc1# copy the include filecp cudnn.h /usr/local/cuda-6.5/includecp libcudnn* /usr/local/cuda-6.5/lib

4.分配权限，安装

chmod +x installcuDNNR2.sh./installcuDNNR2.sh

至此，CUDA6.5和cuDNNv2环境都已经搭好，可以搭建tensorflow

阅读全文

0 0