环境配置
运行环境需和打包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。
- 下载NVIDIA apex源码。
git clone https://github.com/NVIDIA/apex
- 进入apex目录。
cd apex
- 安装。
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
父主题: IVFast训练接口