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