一、Python Error
在 Pytorch 读取参数时,报错 RuntimeError: cuda runtime error (10) : invalid device ordinal。
二、解决方法
造成这个错误的原因主要是本地只有一个 GPU (GPU:0),而程序中使用 GPUs:1。
因此,在程序中找到定义 device 的地方,我的如下:
将其中 "cuda:1" 修改为 "cuda:0",程序将不在报错。
或者
你的配置信息在yaml中:修改对应部分即可
在 Pytorch 读取参数时,报错 RuntimeError: cuda runtime error (10) : invalid device ordinal。
造成这个错误的原因主要是本地只有一个 GPU (GPU:0),而程序中使用 GPUs:1。
因此,在程序中找到定义 device 的地方,我的如下:
将其中 "cuda:1" 修改为 "cuda:0",程序将不在报错。
或者
你的配置信息在yaml中:修改对应部分即可