找回密码
 立即注册
搜索
热搜: 活动 交友 discuz
查看: 376|回复: 1

这个是在用CPU跑吗?

[复制链接]

2

主题

3

回帖

41

零食

入门显卡4G

积分
46
发表于 2024-3-15 17:37:58 | 显示全部楼层 |阅读模式

问题是不是cuda版本的问题,

当前cuda版本是12.1
onnxruntime 1.17.1
python 3.10

H:\AI\sd_wav2lip_68\python310\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py:65: UserWarning: Specified provider 'CUDAExecutionProvider' is not in available provider names.Available providers: 'CPUExecutionProvider'

回复

使用道具 举报

2

主题

3

回帖

41

零食

入门显卡4G

积分
46
 楼主| 发表于 2024-3-15 17:49:26 | 显示全部楼层
程序依赖onnxruntime-gpu 1.15.1版本,
卸载后重装,问题解决,现在是gpu在跑了
pip uninstall onnxruntime onnxruntime-gpu
pip install onnxruntime-gpu==1.15.1
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

QQ|Archiver|手机版|小黑屋|圈圈AI吧

GMT+8, 2024-10-5 11:14 , Processed in 0.118637 second(s), 18 queries .

Powered by Discuz! X3.5

© 2001-2024 Discuz! Team.

快速回复 返回顶部 返回列表