怎样开启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,让速度快一些,有谁知道嘛?