AttributeError: module ‘torch.cuda‘ has no attribtue ‘amp‘ 问题解决

• AttributeError: module 'torch.cuda' has no attribtue 'amp' 问题解决

之前没有使用过apex，所以使用apex的时候，发现报了一条错误。

AttributeError: module 'torch.cuda' has no attribtue 'amp'

经过不断的尝试，终于找到了问题的原因。原因在于torch.cuda.amp是在torch1.6版本之后才引入的，而我所使用的是torch1.4版本，自然就会报错。
我的是使用下面命令，安装1.7.1版本后问题解决。具体自己的对应cuda与版本需要查看官网pytorch进行选择对应的安装。

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch