PyTorch apex库安装(Linux系统)

457 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

系统设置

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安装。