ubuntu下安装配置 tensorflow-gpu

配置环境之前要做的工作是知道自己的显卡信息，然后再去官网下载对应版本NVIDIA版本驱动

1、查看显卡信息

nvidia- smi

2、下载对应版本显卡驱动

NVIDIA驱动下载链接 https://www.nvidia.cn/Download/index.aspx?lang=cn

3、禁用开源nouveau驱动

1）打开终端，先删除旧的驱动：

sudo apt-get purge nvidia*

2）禁用自带的 nouveau nvidia驱动

lsmod | grep nouveau

3）创建一个文件

sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

4）添加如下内容：

blacklist nouveau
options nouveau modeset=0

5）保存文件退出，执行下面命令更新一下

sudo update-initramfs -u

6）重启电脑，打开终端再次输入命令

lsmod | grep nouveau

如果发现什么都没显示，那么说明禁用nouveau 成功了，

4、安装NVIDIA显卡驱动

1）首先按Ctrl+Alt+F1进入控制台，输入用户名密码登陆，执行下面命令关闭图形界面

sudo service lightdm stop

然后cd切换到存放显卡驱动文件的路径，运行安装文件：

sudo sh NVIDIA-Linux-x86_64-390.42.run --no-opengl-files

必须加上 --no-opengl-files ,否则会出现循环登录（登录之后再退出到登录界面）

5、网上查找tensorflow-gpu所对应的cuda和cudnn版本关系

tensorflow-gpu 1.4版本为分界线，需要cuda8 和 cudnn6，