Tensorflow GPU for Ubuntu 16.04

保存安装GPU版的TF一直是一大坑

前前后后搞上一个礼拜也不是不可能

准备好足够的耐心

我的机子:

CPU:I7-4790K @4.0Ghz

GPU:Nvidia GeForce GTX 960

RAM:16GB DDR3

OS:ubuntu 16.04 LTS

1、安装准备:

显卡驱动:官网下载(.run格式)

cuda 8.0 (.run 格式)

cudnn 5.0

2、驱动安装

Nvidia的驱动与Ubuntu自带都nouveau有冲突,所以需要先禁用他

在底部添加

重新生成 kernel initramfs并重启:

重启后禁用lightdm

CTRL+ALT+F1打开命令行,输入用户名密码登录

如果安装过程中提示安装密钥失败,请至BIOS禁用security boot/CSM

如果安装成功,重启后输入命令会显示如下内容

3、安装cuda

当询问是否需要安装驱动请输入N

安装结束后会提示未完全安装,不用理会因为我们在装驱动的时候选了No

安装成功后在开始菜单输入nvidia应该会显示3个应用

4、安装cudnn

解压缩,将文件夹与/usr/local/cuda合并即可

5、环境变量和其他

在末尾加入

保存退出

同样的

 

在末尾加入

保存退出

安装libcupti-dev library

6、安装python和Tensorflow

7、测试

 

“Tensorflow GPU for Ubuntu 16.04”的2个回复

发表评论

电子邮件地址不会被公开。 必填项已用*标注