gpuopenanalytics / pynvml

Provide Python access to the NVML library for GPU diagnostics
BSD 3-Clause "New" or "Revised" License
205 stars 31 forks source link

Support for WSL (Windows Subsystem for Linux) #26

Open jakirkham opened 3 years ago

jakirkham commented 3 years ago

Opening this issue to track WSL (Windows Subsystem for Linux). Currently this is not supported as NVML is not supported on WSL (based on reading this note from the NVIDIA WSL docs).

  1. NVIDIA Management Library (NVML) APIs are not supported. Consequently, nvidia-smi may not be functional in WSL 2.

However when NVML is supported on Windows, it would be nice to try using/testing pynvml on Windows.

For more general RAPIDS support of WSL, please see issue ( https://github.com/rapidsai/cudf/issues/28 ).

pentschev commented 2 years ago

It seems that NVML is supposed to be supported since 510.06.

pentschev commented 2 years ago

Actually, 510.06 is supposed to support NVML within docker containers in WSL, but NVML was initially supported in WSL in 465.42.

jakirkham commented 2 years ago

@kenhester, do you know if there is anything special that needs to be done on the PyNVML side to support WSL? Or should things just work there?

pentschev commented 2 years ago

cc @charlesbluca for vis

kenhester commented 2 years ago

I believe there are many conditions to be met to enable GPU support in WSL2. The docs : https://docs.nvidia.com/cuda/wsl-user-guide/index.html#wsl2-system-requirements

Otherwise, once GPU support is enabled I would expect pynvml to work, though I have not tried it yet.

Ken Hester Solutions Architect Manager | NVIDIA Global Energy Sectorhttps://www.nvidia.com/en-us/industries/energy/ +1.832.216.8736<tel:+1.832.216.8736> @.**@.>

[fb]https://www.linkedin.com/groups/6785068/ [fb]https://www.facebook.com/NVIDIADataCenter/ [fb]https://twitter.com/NVIDIADC

From: Peter Andreas Entschev @.> Date: Thursday, September 23, 2021 at 12:32 PM To: gpuopenanalytics/pynvml @.> Cc: Ken Hester @.>, Mention @.> Subject: Re: [gpuopenanalytics/pynvml] Support for WSL (Windows Subsystem for Linux) (#26)

cc @charlesblucahttps://github.com/charlesbluca for vis

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHubhttps://github.com/gpuopenanalytics/pynvml/issues/26#issuecomment-926016208, or unsubscribehttps://github.com/notifications/unsubscribe-auth/AETKIVOWQHVRKGUVAUTQDSDUDNQD5ANCNFSM4PFAYMHQ. Triage notifications on the go with GitHub Mobile for iOShttps://nam11.safelinks.protection.outlook.com/?url=https%3A%2F%2Fapps.apple.com%2Fapp%2Fapple-store%2Fid1477376905%3Fct%3Dnotification-email%26mt%3D8%26pt%3D524675&data=04%7C01%7Ckhester%40nvidia.com%7C13f54a4821814db8dfe008d97eb8284a%7C43083d15727340c1b7db39efd9ccc17a%7C0%7C0%7C637680151681468104%7CUnknown%7CTWFpbGZsb3d8eyJWIjoiMC4wLjAwMDAiLCJQIjoiV2luMzIiLCJBTiI6Ik1haWwiLCJXVCI6Mn0%3D%7C1000&sdata=xysIkMTJUPpB3Gn29wjWoKpsjFo8WbJjVwcZ0h%2BpzXo%3D&reserved=0 or Androidhttps://nam11.safelinks.protection.outlook.com/?url=https%3A%2F%2Fplay.google.com%2Fstore%2Fapps%2Fdetails%3Fid%3Dcom.github.android%26referrer%3Dutm_campaign%253Dnotification-email%2526utm_medium%253Demail%2526utm_source%253Dgithub&data=04%7C01%7Ckhester%40nvidia.com%7C13f54a4821814db8dfe008d97eb8284a%7C43083d15727340c1b7db39efd9ccc17a%7C0%7C0%7C637680151681478097%7CUnknown%7CTWFpbGZsb3d8eyJWIjoiMC4wLjAwMDAiLCJQIjoiV2luMzIiLCJBTiI6Ik1haWwiLCJXVCI6Mn0%3D%7C1000&sdata=1UVKv8ZgzJ1YwbsR8S5zJCrSQ3BvchZsHZiZYoAndeE%3D&reserved=0.

lmeyerov commented 2 years ago

Also: https://github.com/gpuopenanalytics/pynvml/issues/42