分开安装驱动和cuda
1、安装显卡驱动
(1)方法一从官网下载驱动编译
NVIDIA的中文官网为nvidia显卡驱动
1)打开终端输入以下命令删除旧驱动
sudo apt-get purge nvidia*
2)禁用自带的 nouveau nvidia驱动
首先利用命令行打开
sudo gedit /etc/modprobe.d/blacklist.conf
在文件末端加入
blacklist nouveau
options nouveau modeset=0
然后更新
sudo update-initramfs -u
最后重启
终端下输入命令 lsmod | grep nouveau 查看是否被禁用
3)下载好对应gpu版本.run 形式的显卡驱动
图形化界面按住ctrl+alt+(f1~f6的其中一个)进入命令行模式输入用户名和密码,然后在输入
sudo service lightdm stop
4)给驱动run文件赋予执行权限:
sudo chmod a+x NVIDIA-Linux-x86_64-396.18.run
然后开始安装
sudo ./NVIDIA-Linux-x86_64-396.18.run
5)挂载显卡驱动
modprobe nvidia
6)检查是否安装成功
nvidia-smi
如下图则成功
(2)添加官方的ppa源
1)快捷键ctrl+alt+T打开命令终端输入
sudo add-apt-repository ppa:graphics-drivers/ppa
2)然后刷新软件库并安装最新驱动
sudo apt-get update
3)使用终端命令查看Ubuntu推荐的驱动版本
ubuntu-drivers devices
4)采用apt-get命令在终端安装
sudo apt-get install nvidia-430 nvidia-settings nvidia-prime
5)重启系统并验证 打开终端并输入
nvidia-smi
二、安装cuda和cudnn
cuda的官方网址和cudnn的官方网址,cudnn的下载需要注册账号。
(1)安装cuda
1)验证cuda的安装条件
首先,在终端输入 lspci | grep -I nvidia 查看你的gpu版本信息
接着,验证是否安装gcc,终端输入 gcc –version ,如果未安装,则输入sudo apt-get install build-essential进行安装
最后,验证系统是否安装了kernel header和 package development
终端输入 uname –r 查看内核 接着输入sudo apt-get install linux-headers-$(uname -r)
即 sudo apt-get install linux-headers-4.18.20-25-generic
就可以安装对应kernel版本的kernel header和package development
2)安装cuda
终端输入 sudo sh 你下载的cuda.run
然后一直按住回车键,直到服务条款显示到100%。除了driver选项选择no其他都选择yes。
3)设置环境变量
在终端输入sudo gedit ~/.bashrc
在文本的最后输入
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
目录要对应自己的cuda版本
4)然后source ~/.bashrc 激活环境
5)在终端输入nvcc -V出现下面画面则安装成功
(2)安装cudnn
1)下载对应你cuda版本的cudnn
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
出现以下信息则表明安装成功