menkaiyuan
menkaiyuan
试试这个参数--one_shot
> 提示语有,因为版本不匹配,cuda和torch的版本对不上。 > > 执行如下命令安装指定版本的torch和cuda `pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html` > > 然后在虚拟环境中输入python,进入python的cli模式,输入如下代码,检测版本修改后gpu是否可以用 import torch torch.**version** torch.cuda.is_available() --------------- 我这边使用的rtx4090,然后我运行你的命令提示都是true,并且 我的其他的项目都在正确的使用cuda,但是在这个项目codeformer上一直在用的cpu运行,然后速度非常非常慢,单独处理一张图片预计5-10分钟,您能帮助分析一下吗
 我的命令运行后 就一直保持这样,直到很久以后才开始进行下一步的状态变化