# GPU Utilization is 0 Despite JAX Occupying GPU Memory

**URL:** <https://discourse.scverse.org/t/gpu-utilization-is-0-despite-jax-occupying-gpu-memory/3332>\
**Category:** Help\
**Tags:** gpu\
**Created:** [November 8, 2024, 3:46am UTC](https://discourse.scverse.org/t/gpu-utilization-is-0-despite-jax-occupying-gpu-memory/3332 "2024-11-08T03:46:47Z")\
**Posts on this page:** 2\
**Page:** 1

<div class="post-metadata">

**Author:** ![zoey](https://avatars.discourse-cdn.com/v4/letter/z/b4bc9f/32.png) [@zoey](https://discourse.scverse.org/u/zoey)\
**Post date:** [November 8, 2024, 3:46am UTC](https://discourse.scverse.org/t/gpu-utilization-is-0-despite-jax-occupying-gpu-memory/3332/1 "2024-11-08T03:46:47Z")

</div>

Hello,

I’m running a model using JAX, and I see that it occupies GPU memory, but the GPU utilization remains at 0%. The following message appeared during the execution:

 ![image](https://canada1.discourse-cdn.com/flex035/uploads/forum11/original/2X/f/fcbcce691c9d5e6e63cdec3d598f90a342929087.png)  
Could someone help me understand why the GPU is not being utilized despite the memory being allocated? What could be causing this issue, and how can I ensure that JAX is properly using the GPU for computation?

Thank you in advance for your help!

---

<div class="post-metadata">

**Author:** ![cane11](https://yyz1.discourse-cdn.com/flex035/user_avatar/discourse.scverse.org/cane11/32/241_2.png) [@cane11](https://discourse.scverse.org/u/cane11)\
**Post date:** [November 27, 2024, 6:42am UTC](https://discourse.scverse.org/t/gpu-utilization-is-0-despite-jax-occupying-gpu-memory/3332/2 "2024-11-27T06:42:27Z")

</div>

Hi. Are you sure that the GPU is not being used? Could you increase the batch size to 4096 to see if this actually gets GPU utilization up.