requirements
- Ubuntu 16.04
- python 2.7
- Flask
- tensorflow GPU 版本
安装nvidia driver
经过不断踩坑的安装,终于google到了靠谱的方法,首先检查你的NVIDIA VGA card model
sudo lshw -numeric -C display
可以看到你的显卡信息,比如我的就是 product: GM107M [GeForce GTX 950M] [10DE:139A],然后去NVDIA driver search page搜索你的显卡需要的驱动型号,页面如下:
下面是我的电脑对应的驱动版本
LINUX X64 (AMD64/EM64T) DISPLAY DRIVER
Version: 375.20
Release Date: 2016.11.18
Operating System: Linux 64-bit
Language: English (US)
File Size: 72.37 MB
从搜索的结果页面看到,我的驱动版本应该是375.20,为了再次确认一遍,你还可以使用这个命令查看你可以使用的驱动:
ubuntu-drivers devices
结果显示和搜索到的驱动版本一样,推荐也是375
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
vendor : NVIDIA Corporation
model : GM107M [GeForce GTX 950M]
modalias : pci:v000010DEd0000139Asv000017AAsd0000380Bbc03sc02i00
driver : nvidia-367 - third-party free
driver : nvidia-375 - third-party free recommended
driver : nvidia-364 - third-party free
driver : nvidia-358 - third-party free
driver : xserver-xorg-video-nouveau - distro free builtin
driver : nvidia-370 - third-party free
== cpu-microcode.py ==
driver : intel-microcode - distro non-free
好了,终于可以安装对应的驱动了,使用以下命令
version: 375
sudo apt-get install nvidia-375
//你自己的版本
//version : xxx
//sudo apt-get install nvidia-xxx
什么,安装很慢,找不到包?更换一下软件源,这个自己google怎么更换,最简单的就是图形界面里面找到System->settings->Software&Updates,然后换一下源,比如阿里云或者中科大(我突然不能链接中科大镜像了,真实坑),然后再执行一下命令
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev
安装完成之后,重启电脑,驱动应该就完成了!你可以在dashboard上搜索nvidia,看到像 NVIDIA X Server Settings的东西,就说明安装驱动成功了,接下来就是安装cuda8了