安装方法
git clone https://github.com/NVIDIA/apex
cd apex
python setup.py install
错误提示,如果出现下列类似报错,请注意你的pytorch以及torchvision版本!!
ImportError: cannot import name 'container_abcs' from 'torch._six' (/home/rjw/anaconda3/envs/ChangeFormer/lib/python3.8/site-packages/torch/_six.py)
我的版本,亲测不会报错
torch=1.13.1
torchvision=0.14.1