-
Notifications
You must be signed in to change notification settings - Fork 170
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
ImportError: No module named 'upfirdn2d_plugin' #5
Comments
What version of cuda do you have? |
|
@rotemtzaban same issue, this is my output
|
StyleGAN2 requires having CUDA toolkit 11.0 or later. I will update the requirements to reflect this. |
@sam598 |
@rotemtzaban I updated pytorch version 1.8 to 1.10 (cuda 11.1) and it worked. |
@rotemtzaban I have an RTX 3090 and just upgraded from CUDA 11.0 to 11.3. I am getting the same error with both versions of CUDA.
I also uninstalled and reinstalled torch to 1.10.2 after installing CUDA and cudnn 11.3 |
|
@tg-bomze that did the trick! |
I got this warning...
The text was updated successfully, but these errors were encountered: