-
import jax
jax.random.PRNGKey(0)After I run these two lines of code, I found my GPU memory increases 10GB? It is normal? |
Beta Was this translation helpful? Give feedback.
Answered by
hawkinsp
Jun 7, 2021
Replies: 1 comment 1 reply
-
|
Yes, this is expected and working as intended. See https://jax.readthedocs.io/en/latest/gpu_memory_allocation.html |
Beta Was this translation helpful? Give feedback.
1 reply
-
|
I see, thank you. |
Beta Was this translation helpful? Give feedback.
Answer selected by
hbb1
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Yes, this is expected and working as intended. See https://jax.readthedocs.io/en/latest/gpu_memory_allocation.html