如果你使用的是 PyTorch 的最新版本,那么你需要使用 torch.cuda.amp.autocast() 来替代 amp.initialize()。这是因为最新版本的 PyTorch 中 amp 模块已经更新为 torch.cuda.amp。
如果你仍然希望使用 amp.initialize(),你需要使用 PyTorch 1.7 或更早的版本。但是,这并不推荐,因为这些旧版本可能不包含许多新功能和改进。
还有一种可能是你没有安装 torch.cuda.amp 模块。你可以通过在命令行中运行 "pip install torch==1.7.0 torchvision==0.8.1"