PyTorch apex库安装(Linux系统)

系统设置

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

你可能感兴趣的:(Linux,#,apex库:混合精度,linux,python,cuda,深度学习,pytorch)