search

dattalab / keypoint-moseq

https://keypoint-moseq.readthedocs.io
Other
64 stars 26 forks source link

GPU not being utilized during model training #99

Closed ryanschwark closed 10 months ago

ryanschwark commented 10 months ago

Hi all,

Recently we've discovered that our PC is likely not utilizing our GPU when model training, causing the kernel to die when model training (see attached file). GPU not being utilized

I was wondering, what might be causing the GPU to not be utilized? My PC specs are: Windows 11 Pro OS build 22000.1936, with a 12th Gen Intel(R) Core(TM) i7-12700K, 3.60 GHz, and 64.0 GB installed RAM. I installed keypoint-MoSeq using pip, and have also downloaded the appropriate CUDA 11.1 / cuDNN 8.2.

What might be causing the issue? Thanks so much!

calebweinreb commented 10 months ago

What makes you say its not being utilized? It looks like 98% of the memory is being used.

ryanschwark commented 10 months ago

Hi,

Ahhh OK I see, that's my bad. You're right, looks like it is using 98% of the GPU!

Thanks!