16.04 Nvidia cuda cudnn

一、安装NVIDIA驱动
1.下载驱动文件
去官网下载和自己的显卡适配的驱动文件,是个.run文件。贴个下载地址:https://www.nvidia.cn/Download/index.aspx

NVIDIA-Linux-x86_64-410.57.run

(1)删除原有驱动(可选):$ sudo apt-get remove --purge nvidia*

(2)禁用nouveau,安装NVIDIA需要把系统自带的驱动禁用
打开文件:sudo gedit /etc/modprobe.d/blacklist.conf
在文本最后添加以下内容:

blacklist nouveau
然后保存退出
执行:sudo update-initramfs -u
重启,命令行输入:lsmod | grep nouveau
没有任何输出的话就代表禁用成功。

(3)安装NVIDIA驱动程序(准备另一台电脑,用来看教程或者是查找需要输入的命令)
禁用图形界面服务:sudo service lightdm stop
按Ctrl+Alt+F1进入命令行,输入用户名密码登录(通过Ctrl+Alt+F7可返回界面)
下载的是.run文件,首先给文件赋予执行权限,首先cd到下载目录:(重要的事情说三遍:注意参数、注意参数、注意参数)
打开所在的文件夹:cd ~/download/ (这里应该写你当时把.run文件下载到的文件夹的路径,一般默认是下载文件夹,如果你在这里出现错误,比如找不到文件,或者路径是中文的,可以尝试把.run文件移动到一个home下,这样可以直接打:cd /home/你的用户名/ 这样就可以打开了。)
然后输入:sudo chmod a+x NVIDIA-Linux-x86_64-410.57.run (敲自己下载的文件名字)
这是不会有什么输出。
接着安装:sudo ./NVIDIA-Linux-x86_64-410.57.run –no-opengl-files

no-opengl-files 只安装驱动文件,不安装opengl文件。这个参数最重要
–no-x-check 安装驱动时不检查X服务
–no-nouveau-check 安装驱动时不检查nouveau
后面两个参数可不加。
接着就是一直选择它默认的那个选项就好。
然后挂载驱动: modprobe nvidia
最后查看是否已经安装好了:nvidia-smi

出现这种界面,说明已经安装成功。可以打开图形界面了:sudo service lightdm start。后面的安装直接在终端即可。

二、安装cuda 10.0
安装和自己显卡的适配版本,也可以看上图中显示的和自己安装的NVIDIA适配的cuda version(上图右上角)。我选择的是cuda 10.0。
(1)下载文件
官网地址:https://developer.nvidia.com/cuda-zone 选择适合自己的版本
在这里插入图片描述下载到一个确定的文件夹里,一般默认在“下载”里。
(2)安装

sudo chmod a+x cuda_10.0.130_410.48_linux.run // 获取权限

  sudo sh cuda_10.0.130_410.48_linux.run

因为我是独立安装的NVIDIA 程序,所以进行上面的选择,安装完成后会出现个提醒,这是因为我在安装CUDA的时候没有选择安装驱动,提示需要安装驱动,忽略就行。
(3)加入环境路径
打开.bashrc 文件:sudo gedit ~/.bashrc
打开文件后将下面两句话加入进去:

export LD_LIBRARY_PATH=$LDLIBRARY_PATH:/usr/local/cuda-10.0/lib64

export PATH=$PATH:/usr/local/cuda-10.0/bin

export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0

保存关闭后,输入:source ~/.bashrc

(4)终端输入:nvcc --version 会输出CUDA的版本信息

    三、安装Cudnn
    1)下载安装文件

按需求下载cudnn的安装文件:https://developer.nvidia.com/rdp/cudnn-archive

(2)安装cudnn

解压下载的文件,可以看到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*

(3)查看cudnn版本

在终端输入

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

发布了14 篇原创文章 · 获赞 37 · 访问量 9万+

猜你喜欢

转载自blog.csdn.net/CVAIDL/article/details/103206720