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人工答疑支持
爱发电:点击前往
赞(0) 打赏
软件无法使用?点击查看常见问题说明>>

最新发布

ToDesk、青椒云、顺网云、海马云电脑使用体验分享-诺瓦小站

ToDesk、青椒云、顺网云、海马云电脑使用体验分享

最近调试了一些大的AI应用,我那渣渣电脑带不动,又不适合买新电脑,所以网上找了一些高端显卡云电脑作为替代,最近用了几家,如:ToDesk云电脑、青椒云电脑、顺网云电脑、海马云电脑,各有优劣吧,这里总结分享一下使用体验。 云电脑对比分析 海马...

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

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

支付宝扫一扫

微信扫一扫