Zhendong-Wang / Diffusion-GAN

Official PyTorch implementation for paper: Diffusion-GAN: Training GANs with Diffusion
MIT License
616 stars 65 forks source link

No module named 'upfirdn2d_plugin' #20

Open sahilqure opened 1 year ago

sahilqure commented 1 year ago

Can you please upload the requirements.txt file.

Zhendong-Wang commented 1 year ago

Hi, I have uploaded a environement.yml for you to create the running env. You could see more requirements there. I also modified something in the torch_utils folder to make the code work for higher version of Pytorch.

sahilqure commented 1 year ago

This issue still exists!

Zhendong-Wang commented 1 year ago

The environment.yml works fine on my Linux/Ubuntu machines. If the issue is still there, then probably it is something with your gcc version and cuda version. Please refer to here, https://github.com/NVlabs/stylegan2-ada-pytorch/issues/35

Zhendong-Wang commented 1 year ago

You could also try this Docker file https://github.com/NVlabs/stylegan2-ada-pytorch/blob/main/Dockerfile

sahilqure commented 1 year ago

Please let me know your gcc and cuda version