一、安装显卡驱动
第一步:先禁用官方开源的显卡驱动nouveau
先运行下列命令,如果有输出则代表nouveau正在加载
$ lsmod | grep nouveau
- 新建文件或打开文件
$ gedit /etc/modprobe.d/blacklist-nouveau.conf
2. 在这个文件里面添加这两行,保存退出
modeset
3. 输入以下命令,并重启
$ sudo update-initramfs -u
$ reboot
重启后运行下列命令,如果无输出则代表nouveau已被禁用
$ lsmod | grep nouveau
第二步:安装显卡驱动
重启电脑,进入登录界面的时候,不要登录进入桌面(否则可能会失败,若不小心进入,请重启电脑),直接按Ctrl+Alt+F1进入文本模式(命令行界面),登录账户。
- 输入以下命令,关闭图形化界面
$ sudo service lightdm stop
2. 切换到cuda安装文件的路径:
$ cd /home
$ sudo sh 驱动文件 -no-x-check -no-nouveau-check -no-opengl-files
-no-x-check安装驱动时关闭x服务
-no-nouveau-check 安装驱动时禁用Nouveau
-no-opengl-files 安装时只装驱动文件,不安装Opengl
上述三个必须添加,不然可能导致循环登录界面
3. 重新开启图形化界面
$ sudo service lightdm restart
4. 同时按住Alt + ctrl +F7,返回到图形化登录界面,输入密码登录。
二、安装CUDA
CUDA和显卡驱动的对应关系如下表,注意一下驱动版本不能太旧。
- 与安装显卡驱动一样,先关闭图形化界面,切换到cuda文件目录,
$ sudo sh cuda_10.2_linux.run
注意事项:不要再安装驱动,X代表选中
如果能够成功登录,则表示不会遇到循环登录的问题,基本说明CUDA的安装成功了。循环登陆解决办法在文后。
2. 开启图形化界面,登陆账户,添加环境变量
$ sudo gedit /etc/profile
添加下面两行,一定不要打错,否则登陆困难!!!
export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64$LD_LIBRARY_PATH
执行
source /etc/profile
三、安装cudnn
下载所需要的cudnn版本,解压文件,并在解压出的cuda文件夹中打开终端,执行如下命令:
$ sudo cp include/cudnn.h /usr/local/cuda/include/
$ sudo cp lib/libcudnn* /usr/local/cuda/lib64/
解决循环登陆问题:
在登陆界面状态下,按Ctrl + Alt + F1,进入TUI,执行
$ sudo /usr/bin/nvidia-uninstall
只需要卸载驱动即可,然后重启,重新安装。
官网详细说明:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.htmldocs.nvidia.com