How can I tell what python script is using my GPU

John Isdell 0 Reputation points
2024-03-15T22:52:31.5033333+00:00

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?

Windows
Windows
A family of Microsoft operating systems that run across personal computers, tablets, laptops, phones, internet of things devices, self-contained mixed reality headsets, large collaboration screens, and other devices.
4,748 questions
Windows Hardware Performance
Windows Hardware Performance
Windows: A family of Microsoft operating systems that run across personal computers, tablets, laptops, phones, internet of things devices, self-contained mixed reality headsets, large collaboration screens, and other devices.Hardware Performance: Delivering / providing hardware or hardware systems or adjusting / adapting hardware or hardware systems.
1,543 questions
Windows 11
Windows 11
A Microsoft operating system designed for productivity, creativity, and ease of use.
8,170 questions
0 comments No comments
{count} votes

2 answers

Sort by: Most helpful
  1. EckiS 821 Reputation points
    2024-03-16T09:45:39.9+00:00

    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/

    0 comments No comments

  2. Ian Xue (Shanghai Wicresoft Co., Ltd.) 29,731 Reputation points Microsoft Vendor
    2024-03-18T08:45:12.1766667+00:00

    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.

    0 comments No comments