网站首页编程语言正文

RuntimeError: CUDA error: invalid device ordinal

作者：香博士更新时间： 2022-02-18 编程语言

在这里插入图片描述
解决方法：
造成这个错误的原因主要是GPU编号原因或者cuda安装版本不匹配

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

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

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

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

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

原文链接：https://blog.csdn.net/weixin_42535423/article/details/109320965