mirror of
https://github.com/TencentARC/GFPGAN.git
synced 2026-04-26 07:42:02 +00:00
How to use GPU #223
Reference in New Issue
Block a user
Delete Branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
Originally created by @DrBlou on GitHub (Aug 1, 2022).
Hi all,
First, its a very nice tool, you did a very good job here !
Sadly, i cant use my RTX3080 GPU to process the images, only the CPU is used, can someone help me with this ?
I can tell my GPU isn't used because of this error :
UserWarning: The unoptimized RealESRGAN is slow on CPU. We do not use it. If you really want to use it, please modify the corresponding codes.
Can someone tell me how to set GFPGAN so it use my GPU ?
Thanks a lot in advance
@elen07zz commented on GitHub (Aug 2, 2022):
You need to install nvidia cuda and then install torch with cuda
conda remove torchvision torchaudio pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
@DrBlou commented on GitHub (Aug 3, 2022):
Thanks a lot it works now !!