How can I tell what python script is using my GPU

Question

I am using Stable Diffusion ComfyUI with a RTX 3060 TI 8gig VRAM. There are times when the program stops working, however, in task manager, I will still see the dedicated RAM taken up, and hear the fans running very fast.

I then try to end all programs associated with python via task manager, but the fans still work hard and the VRAM is full. I have tried using Process explorer to find and kill anything running, that does not work. I am forced to restart the PC. If I do not restart the PC on my own, the system will restart and give a minidump that shows NVLDDMKM.sys had an unknown failure. FAILURE_BUCKET_ID: 0x133_ISR_nvlddmkm!unknown_function. Any idea what I can do to prevent a blue screen and then a crash?

Answer 1

in task manager you can add the additional GPU related columns, sorting by this should give a hint on what process is still using the GPU.

maybe you have the same issue as:
https://www.reddit.com/r/StableDiffusion/comments/1143lvp/stable_diffusion_crashing_my_laptops_gpu/

Answer 2

Hi,

Open Task Manager, switch to the Details tab, right click on the column headers, select columns, and add the Command Line and GPU columns. You can get the script path in the Command Line column and GPU usage in the GPU column of the python.exe process.

Best Regards,

Ian Xue

---

If the Answer is helpful, please click "Accept Answer" and upvote it.

Note: Please follow the steps in our documentation to enable e-mail notifications if you want to receive the related email notification for this thread.