如何在Arm环境安装TensorFlow
收藏回复举报
如何在Arm环境安装TensorFlow
发表于2024-04-18 17:13:15
0 查看

        如何在arm的环境上安装TensorFlow?

1、如果当前环境已安装hdf5,那么我们只需要使用pip install tensorFlow即可

2、如果当前环境没安装hdf5,此时我们使用pip安装TensorFlow通常会有报错如图:

cke_10141.png

那么我们需要先安装hdf5,步骤如下:

1、复制命令执行(--prefix配置安装路径,这个可以修改为自己想要安装的路径)

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

安装hdf5之后,根据--prefix配置的目录执行如下命令:

HDF5_DIR=/usr/include/hdf5 pip install --no-binary=h5py h5py

cke_45647.png

至此,hdf5和h5py已完成安装,可以正常使用pip install tensorflow命令安装TensorFlow了。

可以使用pip安装TensorFlow正常使用

cke_73551.png

按照完TensorFlow后切记,使用pip install protobuf==3.20.0重装一下低版本的protobuf,否则编译单算子工程时会有问题。

本帖最后由 匿名用户2024/04/18 19:07:37 编辑

我要发帖子