AI软件下载
有趣网站推荐及实用软件下载
当前位置：诺瓦小站 问题记录 python 正文

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

2024-09-05 分类：python / 问题记录 评论(0)

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)


AI软件用不了？2元爽玩4090： 立即体验>>

热门大语言模型API免费体验： 立即获取>>

软件催更及1对1人工答疑支持： https://nuowa.net/1806
赞(0) 打赏
软件无法使用？点击查看常见问题说明>>
分享到

相关推荐

最近更新

FunASR语音识别转文字软件区分说话人版-诺瓦小站

FunASR语音识别转文字软件区分说话人版

FunASR是一款优秀的基础语音识别框架，配合阿里通义团队开发的其它语音识别模型，能够实现非常好语音转文字效果，尤其在中文语音识别方面，鉴于有些用户需要对语音识别内容区分说话人，我重新做了一版区分说话人的版本。软件支持多语言识别，但是还是尽...

赞(0)AI音频

热门文章

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

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

支付宝扫一扫

微信扫一扫