RuntimeError: CUDA error: invalid device ordinal

RuntimeError: CUDA error: invalid device ordinal_第1张图片
解决方法:
造成这个错误的原因主要是GPU编号原因或者cuda安装版本不匹配

例如当本地只有一个 GPU (GPU:0),而程序中使用 GPUs:1。

因此,在程序中找到定义 device 的地方,我的如下:

device = th. device ("cuda:1")

将其中 “cuda:1” 修改为 “cuda:0”,程序将不在报错。

device = th. device ("cuda:0")

你可能感兴趣的:(报错处理,cuda,gpu,error,RunTimeError)