Ubuntu16.04下便捷安装tensorflow

前言: 这篇博文记载了Ubuntu16.04下便捷安装tensorflow的方法. 摸索了几天, 发现一种Ubuntu下便捷安装tensorflow的方法, 特此分享. 过程大致分为2步:

1. 安装显卡驱动
2. 安装anaconda, 并conda一个新环境, 装入tensorflow

安装显卡驱动

ubuntu下最方便的安装显卡驱动的方式如下图所示, 系统设置-软件和更新-附加驱动-nvidia410-应用更改

安装完成后需要重启一次, 不出意外打开终端输入nvidia-smi就能看到下图:

可以看到cuda 10.0(安装驱动会自动安装cuda?), 1066的显卡型号, 320MiB的显存占用, 43度的显卡温度, 9W的功耗等信息.

安装tensorflow

首先是安装anaconda, 这个请移步官网, 安装最新版本的即可; 接着conda一个新环境, 且包含最新版tensorflow-gpu, 这里笔者命名为tf_gpu

正常安装需要耗费一些时间, 这里笔者已经装过了, 进入这个环境后, 可以看到其默认安装了 python 3.6.7以及tensorflow-gpu 1.12.0

测试

进入python测试一下, 成功.