Ubuntu的nVidia环境配置初次配置还是很麻烦的,特别是很多版本的问题,本文主要是为了自己之后再次遇到的时候可以顺利快速的安装
首先第一步先要下载cuda10.0,由于官网现在都是默认最新版本,要找之前的版本还得改一下网址
下载cuda安装包:cuda官网下载(https://developer.nvidia.com/cuda-10.0-download-archive)
这里最好选择runfile,至于其他格式的本人也没试过,这里我是下在了/home/Downloads下面
下载结束后我们就要进入终端界面按ctrl+alt+F1进入F1终端,输入用户及密码进行登陆。
一.删除原有的驱动
sudo apt-get purge nvidia*
(原来装过nvidia驱动的需要此步)
如果原来有cuda的需要重新安装的需要输下面两个命令,如果只装驱动或者原来没有装过cuda的就不需要下面两个命令
sudo /usr/local/cuda-9.0/bin/uninstall_cuda_9.0.pl
sudo rm -rf /usr/local/cuda-9.0/
二.禁用自带的nouveau 驱动(一般是第一次安装驱动时需要此步骤)
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
(按i键开始编辑,编辑结束后按esc退出编辑,再按 :wq 保存编辑并退出)
在此命令打开的文本中添加两行内容:
blacklist nouveau
options nouveau modeset=0
三.更新上述文本内容使其生效(一般是第一次安装驱动时需要此步骤)
sudo update-initramfs -u
四.重启系统(一般是第一次安装驱动时需要此步骤)
reboot
五.重新打开终端,确定nouveau已经完成禁用(一般禁用后的图形界面都会变,如果变了的一般都是ok了的,这里只需在图形界面中的窗口终端打开就行)
lsmod|grep nouveau
当执行命令后没有任何内容时说明禁用成功
六.进入F1终端模式并登陆(ctrl+alt+F1)
七.禁用图形界面服务
sudo service lightdm stop
八.进入下载好的目录
cd Downloads
这里进入你自己下载cuda的目录
ls
查看目录下的所有文件
九.到这里就可以进行安装了
sudo sh cuda_10.0.130_410.48_linux.run
执行命令后会有一长串的阅读须知就一直按着回车键直到进度到百分之百,紧接着会有一个是否接受的然后输入accept
接着是安装驱动的输入y,如果只想安装驱动后面的就只输入n和按回车就好了,如果是cuda也要装得那后面就只输入y和回车就好,最后安装好了会出现successful和一些安装的路径。
十.配置环境变量
vim ~/.bashrc
在打开的文本中添加
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
(按i键开始编辑,编辑结束后按esc退出编辑,再按 :wq 保存编辑并退出)
最后使用命令
source ~/.bashrc
使它生效
可以使用命令nvcc -V查看安装的版本信息
到此就nvidia驱动和cuda就安装完成了
下载和安装CUDNN;
进入到CUDNN的下载官网:https://developer.nvidia.com/rdp/cudnn-download ,然点击Download开始选择下载版本,当然在下载之前还有登录,选择版本界面如下,我们选择cuDNN Library for Linux:
下载之后是一个压缩包,如下:
cudnn-10.0-linux-x64-v7.4.2.24.tgz
然后对它进行解压,命令如下:
tar -zxvf cudnn-10.0-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
使用以下两条命令复制这些文件到CUDA目录下:
cp cuda/lib64/* /usr/local/cuda-10.0/lib64/
cp cuda/include/* /usr/local/cuda-10.0/include/
拷贝完成之后,可以使用以下命令查看CUDNN的版本信息:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2