大坑:请先尝试conda install tensorflow-gpu
,比自己编译简单多了,连cuda都不用提前装
今天用anaconda成功安装了tensorflow,心疼自己
20180823 update
请注意tensorflow安装教程页面的官方劝退(如下图),如果你坚持,或者系统环境要求只能自行编译,可以尝试以下方法,否则建议用官方编译版
另外,以下过程除了标明sudo的和安装驱动、cuda等,应当全程不需要管理员权限,否则很可能导致环境不匹配,提示libcublas.so无法加载等问题
声明
本人采坑无数,最终决定将编译过程记录下来分享,允许任何手段的非商业使用,包括但不限于转载、修改、部分引用,但是要求声明出处(即声明来源并给出本博客地址)
环境说明
系统:Ubuntu 16.04.3 安装完之后更新到最新
CPU:Intel x86
显卡:4* Nvidia GeForce GTX 1080 (Driver 390.77)
gcc:5.4.0
cuda:8.0 / 9.0 都适用
cudnn:6.0 for cuda 8.0(5.1提示版本过低), 7.1.4 for cuda 9.0
NCCL:2.2.13
显卡驱动、cuda、cudnn
参见其他博客
下载编译bazel
这里建议自己下载源码编译,因为用deb安装会用到管理员权限,会导致最后的tensorflow编译环境出问题
本节内容参考Bazel 官网
1 安装依赖项
sudo apt-get install build-essential openjdk-8-jdk python zip unzip
2 下载Bazel源码
从github.com下