2019年2月21日
pytorch------cpu与gpu load时相互转化 torch.load(map_location=)
摘要： 将gpu改为cpu时，遇到一个报错： RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU 阅读全文
posted @ 2019-02-21 16:56 小呆丶 阅读(42967) 评论(0) 推荐(4) 编辑