在实验室装GPU driver发现出现各种各样的问题,分享一下我成功的步骤
此时你可能已经装了Nvidia驱动,发现黑屏,
ctrl + alt + F1 首先删除Nvidia所有驱动sudo apt-get autoremove nvidia*
sudo reboot
重启后如果输入密码登陆后又弹回来
执行: sudo rm ~/.Xauthority 再重启
重启后应该能进入界面
sudo ubuntu-drivers devices
你会发现它里面会显示推荐你用nvidia的那一款驱动
sudo apt-get install nvidia-375
进行设置
sudo nvidia-xconfig
对GPUconfig
此时你在你的软件栏里面能看到英伟达的设置界面,但是没有像网上说的prime配置选项
sudo apt-get install nvidia-prime
你需要安装这个就有了
prime-selsect query
这个可以查看你当前的GPU
prime-select +GPU名字
可以命令行下切换GPU,也可以用之前的界面
sudo reboot
最后进行reboot
通过下面的命令可以查看GPU的信息,如果显示出GPU情况,恭喜你的GPU就可以工作啦
nvidia-smi
然后再按照keras.cn 的教程进行配置Cuda和CDNN
再安装tenserflow-GPU 和keras 搞定