Cuda 确实很棒,为什么不使用呢?

作者: usemyhand创建于 2023年9月18日更新于 2023年9月18日

RuntimeError: 尝试在 CUDA 设备上反序列化对象,但 torch.cuda.is_available() 返回 False。如果您在仅支持 CPU 的机器上运行,请使用 torch.load,并设置 map_location=torch.device('cpu') 以将存储映射到 CPU。

内容来源: yunjey/pytorch-tutorial