基础环境:
jetpack 4.4,python3.6, ubuntu18.04
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 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
输入上述命令,可以看到以下画面,即成功