Open EricLeer opened 10 months ago
In my build, I did this: sudo pip3 install -U "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html Unsure how you installed Jax, but if you did not do the above - might be worth a try. This is only a guess, on my part.
Hello, it looks like this may relate to an issue where graphcast was pulling in the wrong tree
library dependency. That should now be resolved as the following PR has been merged: https://github.com/google-deepmind/graphcast/pull/25
would you mind reinstalling graphcast from git and checking if you still see the problem?
Trying to run graphcast with the following command:
but I get the following error:
Running on python 3.10 with the following package versions:
Any idea on what the problem might be? It seems to originate from xarray_jax which is trying to call an attribute that doesnt exist.