Linux系统下CUDA和cuDNN环境配置

科技资讯 投稿 6100 0 评论

Linux系统下CUDA和cuDNN环境配置

本人配置环境:linux服务器,ubantu18,显卡驱动11.0,安装CUDA11.0和cuDNN8.2.1。

一、安装CUDA11.0

2、官网下载,复制官网命令到服务器下载。https://developer.nvidia.com/cuda-toolkit-archive

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}} source ~/.bashrc

5、以下命令激活配置,并使用nvcc -V查看配置效果,如图表示成功。

source ~/.bashrc

 二、安装cuDNN8.2.1

https://developer.nvidia.com/rdp/cudnn-archive

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

至此安装完成。

编程笔记 » Linux系统下CUDA和cuDNN环境配置

赞同 (26) or 分享 (0)
游客 发表我的评论   换个身份
取消评论

表情
(0)个小伙伴在吐槽