E:\PulmonaryNodulesDetection_v8\ultralytics\engine\trainer.py:267: FutureWarning: torch.cuda.amp.GradScaler(args…) is deprecated. Please use torch.amp.GradScaler(‘cuda’, args…) instead.
self.scaler = torch.cuda.amp.GradScaler(enabled=self.amp)

这个警告是由于 torch.cuda.amp.GradScaler 已被弃用,建议使用新的 API torch.amp.GradScaler(‘cuda’, enabled=True)。

解决方案:
您需要将 torch.cuda.amp.GradScaler 替换为 torch.amp.GradScaler,并指定设备 ‘cuda’。

修改代码:
在文件 E:\PulmonaryNodulesDetection_v8\ultralytics\engine\trainer.py 的第 267 行,将以下代码:

self.scaler = torch.cuda.amp.GradScaler(enabled=self.amp)

修改为:
self.scaler = torch.amp.GradScaler(enabled=self.amp)

修改后的代码示例:
# 原代码
self.scaler = torch.cuda.amp.GradScaler(enabled=self.amp)

# 修改为
self.scaler = torch.amp.GradScaler(enabled=self.amp)
这样做可以确保代码与最新的 PyTorch 版本兼容,并且消除这个未来的弃用警告。

总结:
将 torch.cuda.amp.GradScaler 替换为 torch.amp.GradScaler。
此修改确保代码与 PyTorch 的最新版本兼容并消除警告。

声明:本站所有项目资源都可以正常运行,亲测无错!而且我们录制了演示视频,在我们注明的环境版本下,项目运行效果完全和演示视频一致。客服QQ:下载须知