Pytorch中GPU训练好模型CPU下使用
GPU训练模型在CPU下使用
今天我想试试那个SkyAR,但是我没找到GPU的电脑我就想在CPU下使用GPU训练好的模型,使用的时遇到了下面这个问题
`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_lo
原创
2021-02-07 10:00:53 ·
860 阅读 ·
1 评论