pytorch环境下torch.cuda.amp的问题

Question

请问我的pytorch环境配置完成后，显示为true

但是在程序中为什么在pytorch环境下报错，在base环境下就没问题呢

Answer 1

望采纳


如果在使用 Pylance 时收到 "torch.cuda.amp" 未解析的错误提示，可能是因为 Pylance 无法识别 torch.cuda.amp 模块。应该不是cuda本身的报错问题。

你可以试试在使用前先导入一下，并在代码的顶部添加注释，告诉 Pylance 这是一个扩展模块：

# @PYTHON.EXTENSION
import torch.cuda.amp as amp