编译和安装

操作步骤

1. 使用PuTTY工具，以root用户登录服务器。
2. （可选）用户根据实际情况选择是否配置虚拟环境。

   pip install virtualenv
   virtualenv Horovod
   source Horovod/bin/activate

3. 执行以下命令进入安装目录。

   cd /path/to/HOROVOD

4. 执行以下命令安装安装TensorFlow。

   pip install tensorflow-2.4.1-cp37-cp37m-manylinux_2_24_aarch64.whl --trusted-host mirrors.tools.huawei.com -i http://mirrors.tools.huawei.com/pypi/simple

5. 执行以下命令更新更新NumPy。

   pip install -U numpy  --trusted-host mirrors.tools.huawei.com -i http://mirrors.tools.huawei.com/pypi/simple
   python3 -c "import tensorflow as tf; print(tf.__version__); print(tf.sysconfig.get_include()); print(' '.join(tf.sysconfig.get_link_flags())); print(' '.join(tf.sysconfig.get_compile_flags()))"

   回显显示以下信息，表示更新成功。

   

6. 执行以下命令安装PyTorch。

   git clone --depth=1 --recursive  https://github.com/pytorch/pytorch
   cd pytorch
   pip install pyyaml --trusted-host mirrors.tools.huawei.com -i http://mirrors.tools.huawei.com/pypi/simple
   python setup.py install

7. 执行以下命令安装依赖。

   pip install pytest ninja mock --trusted-host mirrors.tools.huawei.com -i http://mirrors.tools.huawei.com/pypi/simple

8. 执行以下命令安装Horovod。

   HOROVOD_WITH_PYTORCH=1 HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITH_MPI=1 pip install --no-cache-dir git+https://github.com/horovod/horovod.git@master --trusted-host mirrors.tools.huawei.com -i http://mirrors.tools.huawei.com/pypi/simple