在ARM64平台下安装TensorFlow

在ARM64平台下安装TensorFlow

1.安装前环境配置

由于安装tensorflow 依赖 h5py ，而 h5py 依赖 HDF5 ，需要先编译安装HDF5 ，否则使用 pip 安装 h5py 会报错

1.1安装HDF5

我们从源码进行安装，对于源码安装都可以参考下面步骤。

• 从官网下载HDF5源码包

wget https://support.hdfgroup.org/ftp/HDF5/releases/hdf5-1.10/hdf5-1.10.5/src/hdf5-1.10.5.tar.gz

• 进入下载目录解压源码包

tar -zxvf hdf5-1.10.5.tar.gz

• 进入解压文件夹，执行配置、编译和安装

cd hdf5-1.10.5/
./configure --prefix=/usr/include/hdf5
make
make install

这里需要注意执行配置的时候最好指定安装目录，方便管理。

• 配置环境变量

打开./bashrc文件内容命令如下

vim ~/.bashrc

在./bashrc内容的最末尾加上下面两行。如果--prefix路径和我不一样，需要改成你安装的hdf5路径

export PATH=$PATH:/usr/include/hdf5/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/include/hdf5/lib

最后source一下使其生效

source ~/.bashrc

需要注意的是在./bashrc添加环境变量。我们遇到的大部分下面ImpoerError找不到文件的问题都可能是没用添加到环境变量。可以先查看下对应的包有没安装，如果安装了还报下面的错误应该把环境变量添加上。

ImportError: hdf5.so.40: cannot open shared object file: No such file or directory

1.2 安装h5py

安装h5py直接使用pip安装即可，也可以用源码安装

安装h5py依赖包,安装h5py

pip install Cython

pip install h5py==2.8.0

 

2.安装TensorFlow

安装Tensorflow由于经常断连，直接先下载编译好的whl包安装

git上有现在做好的aarch64,tensorflow的包：地址如下

tensorflow-on-arm

找到对应的release下载安装，这里我选了tensorflow2.0

用pip安装whl文件

pip install tensorflow-2.0.0-cp37-none-linux_aarch64.whl