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人工答疑支持: https://nuowa.net/1806
赞(0) 打赏
软件无法使用?点击查看常见问题说明>>

最近更新

临时邮箱,一次性匿名gmail.com邮箱

有时候网站需要测试一些功能,比如注册登录等,需要用到邮箱,但是自己的邮箱有限,不能更多的测试效果,这时候就需要一些不同后缀的一次性邮箱,网上有很多这样的一次性匿名邮箱,后缀很多,非常适合测试使用,下面分享几个比较好用的临时邮箱。 Gmail...

检测和删除高相似度图片软件AntiDupl免费下载-诺瓦小站

检测和删除高相似度图片软件AntiDupl免费下载

前几天整理手机图片,手机中拍了大量的图片,很多图片是连拍的,图片的相似度极高,甚至就是一模一样的图片,有时候看相册的时候会删除掉这些图片,有时候忘了删。在电脑上整理备份的时候这些高相似度的图片比较占空间,也会额外增加处理时间,所以想找个图片...

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

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

支付宝扫一扫

微信扫一扫