google-deepmind / google-deepmind/weathernext
Jax cannot access GPU when generating predictions
Open
- Dominant language
- Python
- Stars
- 7.7k
- Forks
- 986
- PR merge metrics
- No merged PRs in 30d
Description
When generating predictions, this error comes up:
`WARNING:jax._src.xla_bridge:No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)`
Using an A100 environment on colab with the stock notebook except with the modifications [mentioned here](https://github.com/google-deepmind/graphcast/issues/14#issuecomment-1816474076).
Contributor guide
Assessment
This issue has not been assessed yet.