Spaces:
Running
on
L4
Running
on
L4
为啥用不起来GPU呢?
#51
by
LycheeX
- opened
同样的代码,app.py和app_2.py都试过了,结果都用不起来gpu,根本没有响应,这是咋回事呢?
这里面的文件咋下载啊?有没有快捷方法啊?
我感觉和GitHub不一致才导致不使用GPU。
完全是懵逼的,不知道bug在哪里。
你的torch版本不对,要安装GPU版本的torch
如我的版本安装命令是:pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
要根据你系统环境等下载对应的版本才行。
参考官网https://pytorch.org/