一、前言

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

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

但是安装失败:

在这里插入图片描述

二、正确的安装方法

2.1 从github上下载

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

2.2 切换到下载后的路径

cd apex

2.3 进行安装

python setup.py install

在这里插入图片描述

下载成功!

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐