Pytorch-UNet icon indicating copy to clipboard operation
Pytorch-UNet copied to clipboard
verified publisher icon milesial

CPU and GPU usage reference

Open mvizosopatino opened this issue 7 months ago • 1 comments

Dear all, I am running Pytorch_Unet. Could anyone tell me whether the usage of the CPU and GPU that I show in the picture attached it is what is expected while running the training?

Thank you

Image

mvizosopatino avatar May 13 '25 09:05 mvizosopatino

I think your CPU usage is fine, but the GPU usage is too low. So try the following: 1. Open the Command Prompt (cmd) 2. Start training 3. Run nvidia-smi or nvidia-smi -lms

Depending on the image size&batch size, in my experience, with that GPU you should get close to 100% utilization if you use 512×512 with a batch size of 8.

taipain avatar Sep 02 '25 14:09 taipain