【环境搭建】源码编译安装TensorFlow&Sonnet

从源码编译安装TensorFlow

1. 安装JDK8

# 按照《TensorFlow实战Google深度学习框架》中的内容，安装bazel之前需要先安装JDK8
sudo apt-get install software-properties-common

# 但是在这一步会报错连接超时，所以我直接跳过JDK8的安装，也成功安装了bazel
sudo add-apt-repository ppa:webupd8team/java

sudo aot-get updata

sudo apt-get install oracle-java8-installer

2. 安装Bazel

# 我的编译安装实际上是从这里开始
# 因为《TensorFlow实战Google深度学习框架》出版已经很久了，里面的版本不再适用，这里安装目前最新版本的bazel
访问bazel的GitHub主页，https://github.com/bazelbuild/bazel
在Releases中下载对应系统的最新版本bazel安装文件
我下载的是bazel-0.24.1-installer-linux-x86_64.sh

# 用chmod命令给下载下来的.sh一个可执行权限
chmod +x bazel-0.24.1-installer-linux-x86_64.sh

./bazel-0.24.1-installer-linux-x86_64.sh --user

3. 安装其他依赖

# 还有一些安装包在前面的工作中已经安装，这里未列出
sudo apt-get install swig
pip install keras
pip install mock

4. 安装TensorFlow

git clone --recurse-submodules https://github.com/tensorflow/tensorflow

# 因为我的电脑上没有GPU，所以关于GPU和CUDA的配置选择NO
./configure

# 如果在上一步中配置了GPU和CUDA，则这步为：bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package

# 执行如下命令会在/tmp/tensorflow_pkg文件夹中生成一个.whl文件
bazel-bin/tensorflow/tools/pip_package /tmp/tensorflow_pkg

# 完成编译安装TensorFlow
sudo pip install /tmp/tensorflow_pkg/tensorflow-1.13.1-cp27-cp27mu-linux_x86_64.whl

pip安装TensorFlow

# cpu版
pip install tensorflow
# gpu版
pip install tensorflow-gpu

安装gpu版本的TensorFlow，注意TensorFlow和CUDA的版本的对应关系

pip安装Sonnet

pip install dm-sonnet

这时import sonnet会报错
根据提示分别进行了下面三种方式尝试解决：

sudo pip install tensorflow-probability

并不能解决问题，报错内容不变，用pip list查看，安装的tensorflow-probability版本为0.4.0

sudo pip install tensorflow-probability>0.4.0

安装成功

sudo pip install tensorflow-probability==0.5.0rc0

尝试降低版本，安装也成功

综上，很多报错的原因可能是版本不匹配

结语

如果您有修改意见或问题，欢迎留言或者通过邮箱和我联系。
如果我的文章对您有帮助，转载请注明出处。