Problem with pytorch in training stylegan3 - process timeout

I faced a problem on training stylegan3 where the terminal stuck at "Setting up PyTorch plugin “upfirdn2d_plugin”… ". I have tried all the methods I found, such as reinstall conda, re-download the whole stylegan3 to retrain the whole network, and pip install ninja etc,. but none of them work. I don’t know if anyone met similar situation before, and how do you solve it? I have attached the screenshot of the problem below. Thank you so much!!!

1 Like