XingangPan / DragGAN

Official Code for DragGAN (SIGGRAPH 2023)
https://vcai.mpi-inf.mpg.de/projects/DragGAN/
Other
35.65k stars 3.45k forks source link

KeyError: 'stylegan2_lions_512_pytorch' #132

Open Sky1014 opened 1 year ago

Sky1014 commented 1 year ago

I've installed all the required packages, along with cuda v12.1. But whenever I tried to run python visualizer_drag_gradio.py in conda, I ran with the error, which states that No CUDA runtime is found, using CUDA_HOME='C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1'. In the last line, it told me that 'stylegan2_lions_512_pytorch' somehow went wrong.

I'm have no idea of what I need to fix. Is there anyone who has encountered similar problems?