RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() i
·
场景:执行代码,报错
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location='cpu' to map your storages to the CPU.
意思:运行时错误:正在尝试反序列化CUDA设备上的对象,但torch.CUDA.is_available()为False。如果您在只有CPU的机器上运行,请使用torch.load和map_location='pu'将您的存储映射到CPU。
解读:torc.CUDA.is_available()为False。GPU未正常运行
可能的原因:
软件:torch,显卡驱动,CUDA。
硬件:GPU。
常见的原因:torch和显卡驱动不匹配
解决办法:
查看电脑设备管理器,GPU硬件正常
pip list 查看torch版本,查看匹配的显卡驱动,显示正常
查看CUDA版本,没有CUDA。
卸载torch,重新安装了包含cuda包的torch版本。
正常运行。
更多推荐
已为社区贡献3条内容
所有评论(0)