代写网站建设合同,wordpress wiki模板,快速seo整站优化排行,建筑设计自考一、Python Error
在 Pytorch 读取参数时#xff0c;报错 RuntimeError: cuda runtime error (10) : invalid device ordinal。
二、解决方法
造成这个错误的原因主要是本地只有一个 GPU (GPU:0)#xff0c;而程序中使用 GPUs:1。
因此#xff0c;在程序中找到定义 devi…一、Python Error
在 Pytorch 读取参数时报错 RuntimeError: cuda runtime error (10) : invalid device ordinal。
二、解决方法
造成这个错误的原因主要是本地只有一个 GPU (GPU:0)而程序中使用 GPUs:1。
因此在程序中找到定义 device 的地方我的如下 将其中 cuda:1 修改为 cuda:0程序将不在报错。 或者 你的配置信息在yaml中修改对应部分即可
