解决报错 ImportError:cannot import ‘torch.cuda.amp‘

报错详解

首先明确的是导入错误，导入错误可能是torch没有安装的原因，而我的torch已经安装好，那么就可能是torch版本的问题。

参考这篇知乎文章 PyTorch的自动混合精度（AMP），知道amp功能在torch=1.6版本发布，而我使用的阿里云天池服务器的torch版本是1.4，并没有此功能，所以需要更新torch版本。

更新指令

pip uninstall torch #先卸载此版本的torch
pip install torch==1.7.1 #只要大于1.6即可