安装TensorFlow

操作步骤

1. 使用PuTTY工具，以root用户登录服务器。
2. 执行以下命令下载TenforFlow源码。

   git clone https://github.com/tensorflow/tensorflow.git

3. GPU版本需要安装cuda-toolkit（请自行安装）。
4. 执行以下命令配置TensorFlow编译选项。

   cd tensorflows-xxx
   ./configure

   GPU版本配置如下：

   

   

5. 执行以下命令编译生成.whl文件。

   bazel clean –expunge

6. GPU版本编译命令。

   bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
   mkdir tensorflow-pkg
   bazel-bin/tensorflow/tools/pip_package/build_pip_package --gpu ./tensorflow-pkg

7. 生成的.whl文件在tensorflow_pkg目录下，执行以下命令安装。

   pip install tensorflow-xxx.whl