CodeFormer
CodeFormer copied to clipboard
Running on CPU now! Make sure your PyTorch version matches your CUDA
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