进入 nx 系统后,输入 sudo apt-get update 更新源
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的版本了

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的版本

免责声明:本文系网络转载或改编,未找到原创作者,版权归原作者所有。如涉及版权,请联系删