中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助

编译和安装

操作步骤

  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