jetson tx2 安装cuda 10.2或者cuda 9全过程


基础环境:
jetpack 4.4,python3.6, ubuntu18.04

1. 检查cuda

首先检查你是否已经安装cuda。进入/usr/local中,如果发现cuda-x.x的文件夹,则说明已经安装cuda,就此打住。
在这里插入图片描述
我的没有,接着往下。

2. 提前安装一些必要软件

因为我的机器里面除了python3和pip3,其余的没有安装,因此,这里我需要全部安装上。

  1. setuptools
sudo -H pip3 install setuptools
  1. cmake
sudo apt-get install python-dev
sudo -H pip3 install utils
sudo apt remove cmake
sudo apt purge --auto-remove cmake
version=3.12
build=3
sudo mkdir ~/Downloads/temp
cd ~/Downloads/temp
sudo wget https://cmake.org/files/v$version/cmake-$version.$build.tar.gz
sudo tar -xzvf cmake-$version.$build.tar.gz
cd cmake-$version.$build
sudo ./bootstrap
sudo make -j4
sudo make install

注意,上面的version=3.12和build=3需要分别输入进命令行中,与其他命令一样输入然后enter。这里指定的是下载cmake 3.12。

上述过程耗时有点长,尤其是最后三步,请记住自己做到哪一步了。

安装完毕之后用cmake --version验证一下,一般是没问题的。
在这里插入图片描述

  1. jtop
sudo -H pip3 install jetson-stats

用来查看机器的实时硬件运行状况。安装后命令行输入jtop即可。

3. 安装jetson使用的cuda和cudnn

这里有两种选择:
1、 参考博客:JETSON TX2 安装 CUDA9.0 及 CUDNN7 超详细(真实亲测) 安装cuda 9和cudnn7

链接:https://pan.baidu.com/s/1c4dSxFXwLhmA2ASYHKkAqw
提取码: c9vk

2、 下载jetpack4.4的cuda10.2以及cudnn

链接:https://pan.baidu.com/s/1kedUoWgh4JfCi3Zy4VnQ8w
提取码:1111

下载该文件,解压,解压后的文件应该是这样的
在这里插入图片描述

3、将该文件夹cuda-10.2上传到jetson的/usr/local/内,如图
在这里插入图片描述
进入cuda-10.2,用下面的命令将targets.tar解压为targets文件夹,

sudo tar xzvf targets.tar

创建/targets/aarch64-linux/lib文件夹和/targets/aarch64-linux/include文件夹的软连接,如下

sudo ln -s /usr/local/cuda-10.2/targets/aarch64-linux/lib lib64

sudo ln -s /usr/local/cuda-10.2/targets/aarch64-linux/include include

4、上传之后cuda-10.2的为普通文件,需要将cuda-10.2文件夹的权限更改成可执行文件,并在/usr/local中创建cuda-10.2的软连接cuda

sudo chmod -R +x /usr/local/cuda-10.2

sudo ln -s /usr/local/cuda-10.2 /usr/local/cuda

sudo ldconfig

5、将cuda地址放进环境变量

修改 ~/.bashrc

sudo vim ~/.bashrc

将如下内容复制进文件

export CUBA_HOME=/usr/local/cuda-10.2
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-10.2/bin:$PATH

保存修改

source ~/.bashrc

7、继续安装一些依赖

sudo apt-get update
sudo apt-get install cuda-toolkit-10-2

8、验证

nvcc --version

输入上述命令,可以看到以下画面,即成功
在这里插入图片描述

评论 24
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值