Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
lishang520 opened this issue May 11, 2023 · 1 comment

Comments

@lishang520
Copy link

我已经让我的环境满足了下面这个三个条件:
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,让速度快一些,有谁知道嘛?

@lishang520
Copy link
Author

@SuperSupeng

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant