鱼的眼泪水知道 发表于 2024-7-6 09:48:38

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 ~/.bashrc4、在末尾添加如下两行,注意里面的路径要换成自己的版本,再按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.tgz3、执行如下四条命令,注意换成自己的版本。
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

总结

至此安装完成。
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

来源:https://www.jb51.net/server/323245n4j.htm
免责声明:由于采集信息均来自互联网,如果侵犯了您的权益,请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容,谢谢合作!
页: [1]
查看完整版本: Linux系统下CUDA和cuDNN环境配置方式