-
Notifications
You must be signed in to change notification settings - Fork 336
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
Failed to run with Jax 0.4.18 #139
Comments
I encountered this error
which was also solved by downgrading to 0.4.16 (thanks @yuehaowang), copypasting it here so it's google-able. |
I was running RawNeRF with the latest Jax 0.4.18 but encountered the error message below after training ~300 iterations:
After downgrading Jax from 0.4.18 to 0.4.16, this error was gone.
I was using CUDA 11.8. I installed Jax via
jax[cuda11_local]
. The installed packages werejax v0.4.18
,jaxlib 0.4.18+cuda11.cudnn86
. Not sure if this is due to conflicts with other packages.The text was updated successfully, but these errors were encountered: