脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 网站技巧 → 服务器 → Linux → Linux下CUDA和cuDNN环境配置

Linux系统下CUDA和cuDNN环境配置方式

 更新时间：2024年06月25日 10:46:19   作者：Akmpfen知行  
这篇文章主要介绍了Linux系统下CUDA和cuDNN环境配置方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

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

总结

至此安装完成。

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

最新评论