Open sahilqure opened 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.
This issue still exists!
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
You could also try this Docker file https://github.com/NVlabs/stylegan2-ada-pytorch/blob/main/Dockerfile
Please let me know your gcc and cuda version
Can you please upload the requirements.txt file.