请问我的pytorch环境配置完成后,显示为true
但是在程序中为什么在pytorch环境下报错,在base环境下就没问题呢
请问我的pytorch环境配置完成后,显示为true
但是在程序中为什么在pytorch环境下报错,在base环境下就没问题呢
望采纳
如果在使用 Pylance 时收到 "torch.cuda.amp" 未解析的错误提示,可能是因为 Pylance 无法识别 torch.cuda.amp 模块。应该不是cuda本身的报错问题。
你可以试试在使用前先导入一下,并在代码的顶部添加注释,告诉 Pylance 这是一个扩展模块:
# @PYTHON.EXTENSION
import torch.cuda.amp as amp