AI软件下载
有趣网站推荐及实用软件下载

Python应用报错RuntimeError: CUDA error: invalid device ordinal设备序号无效

Python应用运行时报错,部分错误信息如下:

Traceback (most recent call last):
File “\gradio_ootd.py”, line 24, in
openpose_model_dc = OpenPose(1)
File “\run_openpose.py”, line 32, in init
torch.cuda.set_device(gpu_id)
File “lib\site-packages\torch\cuda__init__.py”, line 420, in set_device
torch._C._cuda_setDevice(device)
RuntimeError: CUDA error: invalid device ordinal
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

报错提示CUDA设备序号选择错误,我电脑上只有一个显卡,默认设备序号应该是0,但是应用代码里写的是1:

openpose_model_dc = OpenPose(1)

我没有第二个显卡,执行这里显然会报错,将数字1改成0就可以了。

openpose_model_dc = OpenPose(0)

电脑不给力运行不了?试试高配云电脑>>
如需1对1人工答疑支持
知识星球(2月2.5折限时优惠中):扫码加入↓ ↓ ↓
赞(0) 打赏
软件无法使用?点击查看常见问题说明>>

最新发布

ToDesk云电脑运行AI软件整合包使用方法-诺瓦小站

ToDesk云电脑运行AI软件整合包使用方法

我分享的大部分AI软件对电脑配置要求都是比较高的。比如显卡一般都要英伟达8G以上显存,有时CPU和内存也有要求,这类配置比较好用的电脑一般都要8000以上,如果你没有电脑或是预算比较充足,可以直接购买高配电脑,如果已经有电脑了或是预算达不到...

觉得文章对你有帮助就打赏一下作者

非常感谢你的打赏,我将有更多的动力继续提供优质内容,让我们一起创建更加美好的世界!

支付宝扫一扫

微信扫一扫