RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count() is 1.

本是古典 何须时尚 2022-08-30 14:55 244阅读 0赞
  1. 出现这样的原因是因为我用了第二块GPU训练,在另一台机器上加载模型的时候,没有了第二块GPU,这时候就要指定用“cuda:0
  2. torch.load('./modelSubmit/encoder_48.pth.tar',map_location='cuda:0')

然后加载模型就正常了。

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2hoaGhoaGhoaGh3d3d3d3d3d3d3_size_16_color_FFFFFF_t_70

发表评论

表情:
评论列表 (有 0 条评论,244人围观)

还没有评论,来说两句吧...

相关阅读