Pytorch load深度模型时报错：RuntimeError: cuda runtime error (10) : invalid device ordinal

最新推荐文章于 2025-03-25 09:44:32 发布

我的暑假作业没写完

最新推荐文章于 2025-03-25 09:44:32 发布

阅读量1.1w

点赞数 8

文章标签： PyTorch python 深度学习

本文链接：https://blog.csdn.net/yinhui_zhang/article/details/86572232

版权

在服务器上尝试在非训练时使用的GPU上加载PyTorch深度学习模型时遇到RuntimeError: cuda runtime error (10) : invalid device ordinal。错误源于模型保存时包含了GPU信息。解决办法是通过代码将加载的张量转移到CPU，或者在指定GPU上进行。参考链接提供了详细解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题背景和描述：

　　我是在服务器上用显卡２上训练我的模型，但是模型还在继续跑，所以我只能在其他显卡上重新做测试实验看效果的好坏。在pytorch上重新load训练好的深度学习模型时报错：RuntimeError: cuda runtime error (10) : invalid device ordinal at torch/csrc/cuda/Module.cpp:32。

THCudaCheck FAIL file=torch/csrc/cuda/Module.cpp line=32 error=10 : invalid device ordinal
Traceback (most recent call last):
  File "las_test.py", line 35, in <module>
    listener = torch.load(listener_model_path)
  File "/home/zyh/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/serialization.py", line 303, in load
    return _load(f, map_location, pickle_module)
  File "/home/zyh/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/serialization.py", line 469, in _load
    result = unpickler.load()
  File "/home/zyh/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/serialization.py", line 437, in persistent_load
    data_type(size), location)
  File "/home/zyh/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/serialization.py", line 88, in default_restore_location
    result = fn(storage, location)
  File "/home/zyh/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/serialization.py", line 70, in _cuda_deserialize
    return obj.cuda(device)
  F

最低0.47元/天解锁文章