CodeFormer icon indicating copy to clipboard operation
CodeFormer copied to clipboard

Running on CPU now! Make sure your PyTorch version matches your CUDA

Open sitte opened this issue 2 years ago • 11 comments

I have a prolem with running CUDA on GPU. When I'm runnig command: python inference_codeformer.py --bg_upsampler realesrgan --face_upsample -w 0.7 --input_path G:\AI\CodeFormer\results\test1.jpg

i'm getting: inference_codeformer.py:49: RuntimeWarning: Running on CPU now! Make sure your PyTorch version matches your CUDA.The unoptimized RealESRGAN is slow on CPU. If you want to disable it, please remove --bg_upsampler and --face_upsample in command.

I uninstalled CUDA 12 and installed 11.7, I reinstalled pytorch with: conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia but no change.

Can somebody sugest some solution?

Enviroment:

Windows 11, Desktop, GTX 3070
Python 3.10.9
pytorch                   1.13.1          py3.9_cuda11.7_cudnn8_0    pytorch
pytorch-cuda              11.7                 h67b0de4_1    pytorch
pytorch-mutex             1.0                        cuda    pytorch
torchaudio                0.13.1                   pypi_0    pypi
torchvision               0.14.1                   pypi_0    pypi

sitte avatar Mar 11 '23 22:03 sitte