昇腾社区首页
中文
注册

环境配置

运行环境需和打包run包的系统环境以及Python版本保持一致,当前已测试的环境为Linux x86_64,Python 3.7.5版本。

环境依赖库参见如下:

  • numpy(version > 1.13.1)
  • faiss-gpu(version = 1.7.1)
  • torch(cuda version = 10.2)
  • argparse
  • NVIDIA apex

其中numpy、faiss-gpu, torch、argparse直接使用pip install命令即可安装,命令执行参考如下。

pip install numpy faiss-gpu torch argparse

NVIDIA apex安装参考 https://github.com/NVIDIA/apex

  1. 下载NVIDIA apex源码。
    git clone https://github.com/NVIDIA/apex
  2. 进入apex目录。
    cd apex
  3. 安装。
    pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./