ubuntu18.04安装CUDA10.1+cudnn

https://blog.csdn.net/wzjwj/article/details/104309743

虽然安装了好多次,但有时候仍然需要搜索一下教程,所以自己记录下安装过程,避免踩坑。

1.安装显卡驱动

采用最简单的方法,可以避免其他方法出现在登陆界面循环出不来,直接在software&updates->additional Drivers里面选择驱动版本更新(如果没有,需要将下载下载源换乘国内镜像,推荐清华 参考:https://blog.csdn.net/Justdoforever/article/details/90949841)然后Revert

然后打开终端

nvidia-smi
出现下面界面则成功

2.安装CUDA(CUDA各种版本)

下载后执行命令

sudo sh cuda_10.1.105_418.39_linux.run
后续如果有安装驱动的选项,选择不安装

安装完成后添加环境变量

sudo gedit /etc/bash.bashrc
复制下列内容

export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64
然后重新初始化执行该文件使之生效

source /etc/bash.bashrc
sudo ldconfig
重开终端,执行

nvcc -V
出现cuda版本即安装成功。

3.安装cudnn(下载地址)

需要注册才能下载,选择与CUDA对应的cudnn版本 且选择cuDNN Library for Linux(是一个压缩包)

下载好后解压,解压后就是cuda文件夹,将对应文件复制到 cuda目录下

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
--ove
————————————————
版权声明:本文为CSDN博主「小萨1332」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/wzjwj/article/details/104309743

原文地址:https://www.cnblogs.com/shuimuqingyang/p/14986333.html