深度学习apex包安装失败的问题

Posted 2023-03-07 旅途中的宽~

文章目录

• 一、前言
• 二、正确的安装方法
• • 2.1 从github上下载
  • 2.2 切换到下载后的路径
  • 2.3 进行安装

一、前言

在进行论文复现的时候，发现缺了一个重要的包：apex包。

这是NVIDIA开发的基于PyTorch的混合精度训练加速神器，能够增加运算速度，并且减少显存的占用。

但是安装失败：

二、正确的安装方法

2.1 从github上下载

git clone https://github.com/NVIDIA/apex.git

2.2 切换到下载后的路径

cd apex

2.3 进行安装

python setup.py install

下载成功！