系统设置
Linux版本:Ubuntu18.04
cuda:11.1.1
之前用cuda9.1无法安装apex,原因是setup.py是用cuda11生成的,无法向下配置。(大概原因是这样,具体怎么说的我忘了…)因此重装了cuda
安装步骤
$ git clone https://github.com/NVIDIA/apex
打开apex文件,注意一下requirements.txt中的依赖库要求:
cxxfilt>=0.2.0
tqdm>=4.28.1
numpy>=1.15.3
PyYAML>=5.1
pytest>=3.5.1
torch库也需要安装。
安装好之后执行以下步骤:
$ cd apex
$ python3 setup.py install
bash环境中的python是python3.8,因此用的是python3而不是python安装。