本文参考
Jetson xavier nx 安装CUDA10.2,和cudnn8.0_jetson安装cuda_三羊木木的博客-CSDN博客
NX安装cuda和cudnn_nx cuda_viatea的博客-CSDN博客
进入nx系统后,输入 sudo apt-get update 更新源
目录
1 cuda
2 cudnn
1 cuda
sudo apt-get install cuda-toolkit-10-2
过程会久一些,我就截取一头一尾了
输入 sudo gedit ~/.bashrc
将这三行复制到最后
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_ROOT=/usr/local/cuda
然后输入 source ~/.bashrc
之后输入 nvcc -V 就可以看到cuda的版本了
2 cudnn
cudnn的包你需要从 SDKManager 中获得,SDKManager的安装方式在这里有 NX从刷机到更换开机logo_Suyuoa的博客-CSDN博客 4.给NX高一个系统(给nx刷机)
你走到这里就行了,把东西下载好,然后在画红框的目录中找
找到这三个文件
把这三个文件都搞到nx上,然后输入 sudo dpkg -i [包名] ,顺序无所谓
然后输入 cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 就可以查看cudnn的版本