ModuleNotFoundError: No module named 'torch.cuda.amp'

这个错误是因为你的 PyTorch 版本过低，缺少了 torch.cuda.amp 模块。请安装 PyTorch 的最新版本，或者升级到支持该模块的版本。你可以尝试使用以下命令升级 PyTorch：

pip install --upgrade torch torchvision

如果你使用的是 GPU 版本的 PyTorch，还需要检查你的 CUDA 版本是否符合要求。你可以在 PyTorch 官方网站上查看 PyTorch 和 CUDA 的兼容性。