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

news/2025/3/4 8:16:01/

1. 检查cuda

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

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

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

setuptools

sudo -H pip3 install setuptools

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验证一下，一般是没问题的。
在这里插入图片描述

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 lib64sudo 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.2sudo ln -s /usr/local/cuda-10.2 /usr/local/cudasudo 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

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