在PyTorch的最新版中，amp模块已更新，推荐使用torch.cuda.amp.autocast()而非amp.initialize()。旧版本虽然仍支持amp.initialize()，但不建议使用，因为会错过新功能和改进。若遇到未找到torch.cuda.amp，可能需安装特定版本如1.7.0。
如果你使用的是 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"
141
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
