参考 RuntimeError: CUDA error (10): invalid device ordinal - 云+社区 - 腾讯云
一、Python Error
在 Pytorch 读取参数时,报错 RuntimeError: cuda runtime error (10) : invalid device ordinal。
二、解决方法
造成这个错误的原因主要是本地只有一个 GPU (GPU:0),而程序中使用 GPUs:1。
因此,在程序中找到定义 device 的地方,我的如下:
![]()
将其中 "cuda:1" 修改为 "cuda:0",程序将不在报错。
![]()
本文介绍了一种常见的Pytorch运行时错误,RuntimeError: cudartError(10): invalid device ordinal,主要由于程序试图访问不存在的GPU(如cuda:1),在只有一个GPU(cuda:0)的环境中。通过修改代码中设备标识为cuda:0,解决了这个问题。
3万+

被折叠的 条评论
为什么被折叠?



