失眠网,内容丰富有趣,生活中的好帮手!
失眠网 > NX安装cuda和cudnn

NX安装cuda和cudnn

时间:2023-02-16 18:29:02

相关推荐

NX安装cuda和cudnn

1.安装cuda

sudo apt-get updatesudo apt-get install cuda-toolkit-10-2

打开命令行添加环境变量即可

我这里是cuda10.2大家要根据自己的cuda版本去填写路径

vi ~/.bashrcexport 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/cudasource ~/.bashrc

查看是否配置成功

nvcc -V

2.安装cudnn

下载链接,我设的不要钱

/download/viatea/48244611

直接使用 下面的命令安装即可:

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

然后将头文件与库文件复制到cuda目录下:

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

修改文件权限

sudo chmod 777 /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

重新进行软链接,如果和我一样使用cuda10.2,这边就不用动

cd /usr/local/cuda/lib64sudo ln -sf libcudnn.so.8.0.0 libcudnn.so.8sudo ln -sf libcudnn_ops_train.so.8.0.0 libcudnn_ops_train.so.8sudo ln -sf libcudnn_ops_infer.so.8.0.0 libcudnn_ops_infer.so.8sudo ln -sf libcudnn_adv_infer.so.8.0.0 libcudnn_adv_infer.so.8sudo ln -sf libcudnn_cnn_infer.so.8.0.0 libcudnn_cnn_infer.so.8sudo ln -sf libcudnn_cnn_train.so.8.0.0 libcudnn_cnn_train.so.8sudo ln -sf libcudnn_adv_train.so.8.0.0 libcudnn_adv_train.so.8

编译一下

sudo ldconfig

cuDNN版本查询:

cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

安装成功会显示如下

#define CUDNN_MAJOR 8

#define CUDNN_MINOR 0

#define CUDNN_PATCHLEVEL 0

#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#endif /* CUDNN_VERSION_H */

3.安装输入法

sudo apt-get install fcitx-googlepinyin -y

安装完成后,进入Language Support,将最下面的ibus改为fcitx

重启机器,使用reboot

重启后,点击右上角的键盘,左键点击,ConfigureFcitx,进入下面画面

使用ctrl+space切换

如果觉得《NX安装cuda和cudnn》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。