源码安装
提前装好NVIDIA驱动,cuda9.0,cudnn,安装bazel-0.15,这里tensorflow-gpu直接从github上下载源码,
下载地址:https://github.com/tensorflow/tensorflow
解压,进入解压目录
./configure
配置里只设置使用cuda,因为我是在自己目录下安装的cuda,所以需要修改cuda路径
然后对于nccl的配置(如下):因为服务器不联网,所以这里不能默认(默认是从github上下载),nccl安装见另一篇文章https://mp.csdn.net/postedit/83347822
这样设置就可以了
在执行下面的命令进行编译阶段,由于服务器不联网,无法下载各种包
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:bu