Ubuntu16.04安装和卸载CUDA驱动和Cudnn

本文档详细介绍了在Ubuntu系统中如何安装NVIDIA驱动、CUDA和CuDNN的步骤,包括下载相应版本、禁用nouveau驱动、安装与配置过程以及可能出现的问题与解决方法。内容覆盖了从驱动下载到CUDA和CuDNN的安装、环境变量配置,以及后续的卸载流程,为GPU计算环境的搭建提供了全面的指南。
摘要由CSDN通过智能技术生成

安装NVIDIA驱动

1、下载驱动

官网下载地址
NVIDIA驱动高级搜索,可以下载历史版本
根据自己电脑显卡的型号和操作系统,下载对应版本的显卡驱动。
本人选择的是

选项前缀名称选项
产品类型GeForce
产品系列GeForce 10 Series (Notebooks)
产品家族GeForce GTX 1050 Ti
操作系统Linux 64-bit
下载类型Linux 长寿命
语言Chinese (Simplified)

2、禁用nouveau驱动

#步骤一:
sudo vim /etc/modprobe.d/blacklist.conf

'''
在文本最后添加:
blacklist nouveau
options nouveau modeset=0
'''

# 步骤二:
# 更新系统
sudo update-initramfs -u

# 步骤三:验证nouveau是否已禁用
# 重启后,执行以下命令,如果没有屏幕输出,说明禁用nouveau成功:
'一定要重启'

lsmod | grep nouveau

3、安装驱动

sudo chmod a+x NVIDIA-Linux-x86_64-410.93.run
sudo ./NVIDIA-Linux-x86_64-396.18.run -no-x-check -no-nouveau-check -no-opengl-files //只有禁用opengl这样安装才不会出现循环登陆的问题
'''
-no-x-check:安装驱动时关闭X服务
-no-nouveau-check:安装驱动时禁用nouveau
-no-opengl-files:只安装驱动文件,不安装OpenGL文件
'''

4、挂载和检测

挂载Nvidia驱动:

modprobe nvidia

检查驱动是否安装成功:

nvidia-smi

卸载NVIDIA驱动

下面的操作都是在命令行界面进行操作的,通过下面的快捷键进入命令界面,并输入用户ID和password登录:

Ctrl-Alt+F1

执行以下命令禁用X-Window服务,否则无法安装显卡驱动:

sudo service lightdm stop

执行命令卸载原有显卡驱动:

sudo apt-get remove --purge nvidia*
sudo chmod +x NVIDIA-Linux-x86_64-410.93.run
sudo ./NVIDIA-Linux-x86_64-410.93.run --uninstall

安装CUDA

1、下载CUDA

由于部分用户不喜欢下载最新的版本,或者电脑不支持之类的,所以在这里提供各种版本的下载地址。本人下载的是CUDA 10.0
CUDA各种版本下载地址

2、开始安装

sudo sh cuda_10.0.130_410.48_linux.run

开始安装之后,需要阅读说明,可以使用Ctrl + C直接阅读完成,或者使用空格键慢慢阅读。

是否同意条款,必须同意才能继续安装)
accept/decline/quit: accept

(这里不要安装驱动,因为已经安装最新的驱动了,否则可能会安装旧版本的显卡驱动,导致重复登录的情况)
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit: n

Install the CUDA 10.0 Toolkit?(是否安装CUDA 10 ,这里必须要安装)
(y)es/(n)o/(q)uit: y

Enter Toolkit Location(安装路径,使用默认,直接回车就行)
 [ default is /usr/local/cuda-10.0 ]:  

Do you want to install a symbolic link at /usr/local/cuda?(同意创建软链接)
(y)es/(n)o/(q)uit: y

Install the CUDA 10.0 Samples?(不用安装测试,本身就有了)
(y)es/(n)o/(q)uit: n

Installing the CUDA Toolkit in /usr/local/cuda-10.0 ...(开始安装)

3、配置环境变量

gedit ~/.bashrc的最后加上以下配置信息:

export CUDA_HOME=/usr/local/cuda-10.0
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}

最后使用命令source ~/.bashrc使它生效。

4、查看是否安装成功

可以使用命令nvcc -V查看安装的版本信息:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130

5、出现的问题

1)Installation failed. See log at /var/log/cuda-installer.log for details.【暂未找到较好的解决方法】

问题描述:第一次下载的是CUDA 10.2 安装的时候出现下面错误“Installation failed. See log at /var/log/cuda-installer.log for details.”
出现原因:没有找到
解决方法:换了一个版本重新安装

卸载CUDA

'执行的是CUDA自带的卸载脚本,读者要根据自己的cuda版本找到卸载脚本:'
sudo /usr/local/cuda-9.2/bin/uninstall_cuda_9.2.pl

'执行完上面的命令之后,会有一些文件残留,可以执行下面的命令,一起删除'
sudo rm -rf /usr/local/cuda-9.2/

安装Cudnn

1、下载Cudnn

Cudnn官网下载地址
选择cuDNN Library for Linux下载。

下载之后是一个压缩包,比如 “ cudnn-10.2-linux-x64-v7.4.2.24.tgz ”

2、对压缩包进行解压,命令如下:

tar -zxvf cudnn-10.2-linux-x64-v7.4.2.24.tgz 

解压之后可以得到以下文件:

cuda/include/cudnn.h
cuda/NVIDIA_SLA_cuDNN_Support.txt
cuda/lib64/libcudnn.so
cuda/lib64/libcudnn.so.7
cuda/lib64/libcudnn.so.7.4.2
cuda/lib64/libcudnn_static.a

3、拷贝文件

使用以下两条命令复制这些文件到CUDA目录下:

cp cuda/lib64/* /usr/local/cuda-10.0/lib64/
cp cuda/include/* /usr/local/cuda-10.0/include/

4、查看信息,是否安装成功

使用以下命令查看CUDNN的版本信息:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值