安装深度学习框架tensorflow和pytorch需要使用cuda/cudnn(GPU)来加速计算,而安装cuda/cudnn,首先需要安装nvidia的显卡驱动
安装的过程很容易出现驱动冲突,循环登录两个问题,这种问题让人很头疼的,下面来介绍一下如何避免这两个问题的出现并成功安装NVIDIA驱动,以及如何选择适合自己电脑的驱动。
如果你出现循环登陆的界面解决方式如下:
在ubuntu下按ctrl+alt+f1进入命令行界面:输入一下语句
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get install nvidia-384#根据自己的安装版本号进行更改
然后用一下方式重启电脑:
sudo reboot 就解决了循环登录的问题。
如果解决了循环登陆问题或者准备安装这样操作(避免进入循环登陆的安装方式)
1.首先禁用Ubuntu驱动程序nouveau,安装nvidia显卡驱动
禁用命令如下:
a、如果你没有安装若vim执行下面这句(安装了这个的话就跳过这句)
sudo apt-get install vim
b、编辑文件blacklist.conf
sudo vim /etc/modprobe.d/blacklist.conf
在文件打开得文件下插入两行:
blacklist nouveau
option