为啥用不起来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/

Sign up or log in to comment