安装CUDA和cuDNN

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_20965753/article/details/89218205

升级NVIDIA驱动, 安装CUDA!

CUDA历史版本: https://developer.nvidia.com/cuda-toolkit-archive
cuDNN历史版本: https://developer.nvidia.com/rdp/cudnn-archive

一. 卸载原驱动

1.打开终端,先删除旧的驱动
sudo apt-get purge nvidia*
2.禁用自带的 nouveau nvidia驱动
创建一个文件通过命令
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
并添加如下内容:
blacklist nouveau
options nouveau modeset=0
再更新一下:
sudo update-initramfs -u
修改后需要重启系统。

确认下Nouveau是已经被禁用
lsmod | grep nouveau
没有输出什么东西,说明已经成功关闭了

二. 安装最新驱动

安装之前首先得下载NAVIDIA驱动, https://www.nvidia.cn/Download/index.aspx?lang=cn. 下载最新版本的对应显卡的驱动.

  1. 关闭X-window服务
    重启系统至文本模式,也可先进入图形桌面再进入文本模式,再安装下载的驱动就无问题。首先我们需要结束x-window的服务,否则驱动将无法正常安装;关闭X-Window。
    sudo service lightdm stop
  2. sudo ./NVIDIA-Linux-x86_64-390.77.run
    不检查X服务, 选择NO.
  3. 检查
    终端运行确认是否正确安装。
    nvidia-smi
    nvidia-settings

三. 卸载CUDA版本

  1. –purge选项会将配置文件、数据库等删除
    sudo apt-get autoremove --purge cuda
  2. 下载https://developer.nvidia.com/cuda-toolkit-archive, 下载CUDA8.0.
    sudo sh cuda_8.0.61_375.26_linux.run
    启动安装程序,一直按空格到最后,输入accept接受条款
    输入n不安装nvidia图像驱动,之前已经安装过了
    输入y安装cuda 8.0工具
    回车确认cuda默认安装路径:/usr/local/cuda-8.0
    输入y用sudo权限运行安装,输入密码
    输入y或者n安装或者不安装指向/usr/local/cuda的符号链接
    输入y安装CUDA 8.0 Samples,以便后面测试
    回车确认CUDA 8.0 Samples默认安装路径,该安装路径测试完可以删除
    如果出现了Not enough space on parition mounted at /.错误, 可以这样处理, 在安装cuda路径时, 选择如下:
    Enter Toolkit Location
    [ default is /usr/local/cuda-8.0 ]: /home/link/cuda-8.0
    设置Toolkit的地址要和sudo sh cuda_8.0.61_375.26_linux.run --tmpdir=/home/link/cuda-8.0一起使用才可以!!
  3. 配置环境变量
    sudo vi /etc/profile
    添加变量
    export PATH=/usr/local/cuda/bin/: P A T H e x p o r t L D L I B R A R Y P A T H = / u s r / l o c a l / c u d a / l i b 64 / : PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64/: LD_LIBRARY_PATH
  4. 测试

编译并测试设备 deviceQuery:

cd /home/ly/CUDA/NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery
make
./deviceQuery

Result = PASS,说明CUDA安装成功啦。

5. 安装cuDNN6.0

cuDNN还是自行安装, 直接export LD_LIBRARY_PATH=/home/ly/cudnn6.0/cuda/lib64:$LD_LIBRARY_PATH即可, 这里是cuDNN v6.0版本.
如果不添加cuDNN支持, 有些GPU应用会报错.

6. CUDA9.0 + cuDNN V7

sh .run --tmpdir=/home/
安装CUDA9.0后, 配合使用cuDNN V7. 安装cuDNN流程如下:

sudo cp cudnn.h /usr/local/cuda/include/ # 复制头文件
cd …/lib64 # 打开lib64目录
sudo cp lib* /usr/local/cuda/lib64/ # 复制库文件
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* # 给所有用户增加这些文件的读权限

建立软连接:
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.6
sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -s libcudnn.so.6 libcudnn.so

添加环境变量:
sudo gedit /etc/profile
PATH=/usr/local/cuda/bin:$PATH
export PATH
保存后, 创建链接文件

sudo vim /etc/ld.so.conf.d/cuda.conf
按a进入插入模式, 增加下面一行
/usr/local/cuda/lib64
按esc退出插入模式, 按:wq保存退出
最后在终端输入
sudo ldconfig

猜你喜欢

转载自blog.csdn.net/qq_20965753/article/details/89218205
今日推荐