我要评论linux下cuda和cudnn环境配置
本人配置环境：
- linux服务器
- ubantu18
- 显卡驱动11.0
- 安装cuda11.0和cudnn8.2.1
一、安装cuda11.0
1、先多找几篇博客，了解大概的流程，避免踩坑。
2、官网下载，复制官网命令到服务器下载。
cuda toolkit archive | nvidia developer
3、输入命令，再按i键，进入vim的编辑模式。
sudo vim ~/.bashrc
4、在末尾添加如下两行，注意里面的路径要换成自己的版本，再按esc键退出vim编辑模式，并输入 ：wq 保存并退出。
export path=/usr/local/cuda-11.0/bin${path:+:${path}}
export ld_library_path=/usr/local/cuda-11.0/lib64\
${ld_library_path:+:${ld_library_path}}
5、以下命令激活配置，并使用nvcc -v查看配置效果，如图表示成功。
source ~/.bashrc
二、安装cudnn8.2.1
1、官网下载和cuda对应的版本
cudnn archive | nvidia developer
2、解压文件
tar -xzvf cudnn-11.3-linux-x64-v8.2.1.32.tgz
3、执行如下四条命令，注意换成自己的版本。
sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.0/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.0/lib64 sudo chmod a+r /usr/local/cuda-11.0/include/cudnn*.h sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*
4、使用如下命令查看cudnn的版本号
cat /usr/local/cuda-11.0/include/cudnn_version.h | grep cudnn_major -a 2
总结
至此安装完成。
以上为个人经验，希望能给大家一个参考，也希望大家多多支持代码网。
发表评论