datawhalechina / vced

VCED 可以通过你的文字描述来自动识别视频中相符合的片段进行视频剪辑。该项目基于跨模态搜索与向量检索技术搭建,通过前后端分离的模式,帮助你快速的接触新一代搜索技术。

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

怎样开启GPU呢?按照service下的app.py里面的逻辑修改,但是还是没用

lishang520 opened this issue · comments

我已经让我的环境满足了下面这个三个条件:
os.environ.get("DEVICE") == "cuda" and
torch.cuda.is_available() and
os.environ.get("JINA_MP_START_METHOD") == "spawn"
但是当我上传视频开始提取画面的时候,却报错了,提示cuda无法使用,难道还需要修改其他地方嘛?
我的本机是win10,笔记本电脑,本机有6g显卡。我用的wsl里的ubuntu
像开启GPU,让速度快一些,有谁知道嘛?