linux离线安装apex(C++扩展版本和仅Python版本)

文章目录

  • 安装过程
  • 本人没成功的过程


安装过程

在github上https://github.com/nvidia/apex下载压缩包。
linux离线安装apex(C++扩展版本和仅Python版本)_第1张图片
1.仅Python的版本还是很容易成功的:
解压,进入setup.py所在文件夹,安装即可。

unzip apex-master.zip
cd apex-master
python setup.py install

安装成功示例:
在这里插入图片描述
linux离线安装apex(C++扩展版本和仅Python版本)_第2张图片

2.C++扩展版本安装:
apex官方说仅Python版本会变慢,所以不甘心依旧尝试想离线安装完整版本。
①首先按照requirements.txt安装相应的库,否则会报错。

cxxfilt>=0.2.0
tqdm>=4.28.1
numpy>=1.15.3
PyYAML>=5.1
pytest>=3.5.1

可在清华源https://pypi.tuna.tsinghua.edu.cn/simple/下载相应的whl,之后放入linux服务器 pip install XXX.whl进行安装。(离线安装没有在线安装那样方便,所以可能不止安装这几个库,比如pytest需要额外安装pluggy、atomicwrites等库才能安装成功,慢慢安装一定可以成功!)
②类似仅Python版本的安装代码:

unzip apex-master.zip
cd apex-master
python setup.py install --cpp_ext

安装成功!且之前仅Python版本的如下warning会消失

Warning:  multi_tensor_applier fused unscale kernel is unavailable, possibly because apex was installed without --cuda_ext --cpp_ext. Using Python fallback.  Original ImportError was: ModuleNotFoundError("No module named 'amp_C'")
Warning:  apex was installed without --cpp_ext.  Falling back to Python flatten and unflatten.
Warning:  apex was installed without --cpp_ext.  Falling back to Python flatten and unflatten.
Warning:  apex was installed without --cpp_ext.  Falling back to Python flatten and unflatten.
Warning:  apex was installed without --cpp_ext.  Falling back to Python flatten and unflatten.

本人没成功的过程

服务器没网,尝试下载apex-master后,用以下官方安装代码安装失败:

cd apex-master
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

用网上小伙伴的安装代码也没成功(我也不知道为啥加上–cuda_ext会报错且安装不成功…),如下:

cd apex-master
python setup.py install --cpp_ext --cuda_ext

你可能感兴趣的:(linux,pytorch)