场景:执行代码,报错

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版本。

正常运行。

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐