Jetson Agx Xavier 安装cuda10.2和cudnn

1 安装cuda

首先在主目录下创建一个叫cuda-l4t的文件夹

mkdir cuda-l4t

将cuda10.2和cudnn的安装包放在该文件夹下
cuda-l4t

cd ~/cuda-l4t
bash cuda-l4t.sh cuda-repo-l4t-10-2-local-10.2.89_1.0-1_arm64.deb 10.2 10-2

等待安装成功

2 安装cudnn

sudo dpkg -i libcudnn8_8.0.0.180-1+cuda10.2_arm64.deb
sudo dpkg -i libcudnn8-dev_8.0.0.180-1+cuda10.2_arm64.deb
sudo dpkg -i libcudnn8-doc_8.0.0.180-1+cuda10.2_arm64.deb

安装完成后，其实是直接安装到了默认安装路径usr/include和usr/lib下的，因此需要将其拷贝到cuda安装路径下：

sudo cp /usr/include/cudnn.h  /usr/local/cuda/include/
sudo cp /usr/lib/aarch64-linux-gnu/libcudnn*  /usr/local/cuda/lib64/

然后执行下面命令，用重新生成cudnn库的软连接

cd /usr/local/cuda/lib64
sudo chmod +r libcudnn*
sudo rm -rf libcudnn.so libcudnn.so.8
sudo ln -s libcudnn.so.8.1.1 libcudnn.so.8
sudo ln -s libcudnn.so.8 libcudnn.so

sudo ldconfig

链接最好依照实际链接文件
安装完成

查看cuda版本

nvcc -V

查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2