1 简介:
之前在安装可支持版本的tensorflow-gpu时发现,最新的版本只支持到cuda9.0。找了很多网页,想有人分享一下,结果没有找到。后来只能自己编译,中间遇到很多坑再次说一下,以免贴友重蹈覆辙。基本的环境那装我就不介绍,我默认大家的cuda9.2+cudnn7.0+bazel(注意要安装jdk)已安装好。
2 安装步骤
tensorflow源码下载地址 https://github.com/tensorflow/tensorflow
注意:解压文件后,在文件目录下执行
接下来试一下配置的设置:
提醒:NCCL 2.2.13若是未安装,请按下面操作
接下来编译完成,执行:
生成的tensorflow-gpu包在执行完毕后可在/tmp/tensorflow_pkg目录中。
安装验证如下图: