网上有很多教程来解决这个问题:
可能最多的是:
git clone https://www.github.com/nvidia/apex
cd apex
python setup.py install [--cuda_ext] [--cpp_ext]
类似这种模式的,从github上git clone。
介于一些小白(没错就是我)还需要去安装git,git安装又需要教程,所以推荐:
解决方法:
直接到 https://www.github.com/nvidia/apex 网页,也就是上方代码git clone的地址,下载文件夹。
下载完成:
然后直接将里面的文件直接复制到你需要的python项目中即可。
后续更新:
以上操作是因为想使用注释部分代码:
#from apex.parallel import DistributedDataParallel as DDP
#from apex import amp
import torch.amp
from torch.nn.parallel import DistributedDataParallel as DDP
但实际上 apex.amp和apex.parallel.DistributedDataParallel都已经deprecated了,可以直接用torch.amp和torch.parallel.DistributedDataParallel。