一、先安装Nvidia驱动程序
先确定对应的显卡驱动,然后下载
https://www.geforce.cn/drivers
(驱动下载命令
wget -c http://cn.download.nvidia.com/XFree86/Linux-x86_64/410.78/NVIDIA-Linux-x86_64-410.78.run
)
按照以下教程进行
https://blog.csdn.net/yhaolpz/article/details/71375762
或:自动安装命令:第一次使用的时候失败,可能是因为当时没有配好所需gcc和cmake,后来手动安装的时候装上了
ubuntu-drivers autoinstall
禁用 nouveau
配置环境变量
按照第五步关闭桌面服务进入文本模式
(Ubuntu18.04以上命令部分不可用,改为:
停止可视化桌面:
$ sudo telinit 3
之后会进入一个新的命令行会话,使用当前的用户名密码登录
在相应路径下安装NVIDIA驱动(安装文件也可为.sh后缀,如果提示没有权限使用sudo
):
$ bash NVIDIA-Linux-x86_64-384.111.bin
干净的系统安装时需要gcc和cmke:(gcc在后面安装cuda时有版本要求,查看https://docs.nvidia.com/cuda/)
sudo apt install gcc
sudo apt install cmake
在安装结束后,在命令行输入一下命令重启,NVIDIA驱动即可安装成功:
$ sudo reboot
)
安装驱动
安装驱动如果提示GCC版本不对,按照以下教程更换版本
https://blog.csdn.net/u012968002/article/details/71389713
测试,nvidia-smi,出现显卡信息则为成功
二、cuda安装
下载地址,https://developer.nvidia.com/cuda-toolkit-archive
因为安装nVidia驱动的时候安装了最新的版本,而且卸载失败,所以抱着试试看的心态尝试安装cuda10.1,结果安装成功。
在安装时需要gcc和g++,之前没有装g++,安装一下
$ sudo apt install g++
cuda10.1的安装界面与看到的教程不太一样,更加方便,由于之前安装了nVidia的驱动,在安装cuda时不选择安装nVidia驱动以免出错。
三、cudnn安装
在官网下载安装包,需要注册登录才能下载。选择适合自己的版本
下载完成后解压并进入文件夹:
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*
在终端查看CUDNN版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
完成