2021年9月2日
报错：RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1
摘要： 报错：RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1原因：在使用Pytorch加载模型时报错。加载的模型是用两个GPU训练的，而加载模型的电脑只有一个 阅读全文
posted @ 2021-09-02 15:48 小筱痕 阅读(720) 评论(0) 推荐(0)