Skip to content

Moving from pynvml to nvidia-ml-py #151

New issue
New issue
Closed
#490
#245
Closed
Moving from pynvml to nvidia-ml-py#151
#490
#245
Assignees
ncclementi
Labels
build/testBuilding and testing the documentationimprovementImproves an existing functionality
@jakirkham

Description

@jakirkham
jakirkham
opened on Feb 14, 2023

Currently this is making use of pynvml in a few places:

deployment/source/guides/mig.md

Line 33 in 3b6a991

Alternatively, MIG instance names can be obtained programatically using [NVML](https://developer.nvidia.com/nvidia-management-library-nvml) or [PyNVML](https://github.com/gpuopenanalytics/pynvml). Please refer to the [NVML API](https://docs.nvidia.com/deploy/nvml-api/) to write appropriate utilities for that purpose.

deployment/source/cloud/azure/azure-vm-multi.md

Lines 57 to 60 in 3b6a991

import pynvml
pynvml.nvmlInit()
return pynvml.nvmlDeviceGetName(pynvml.nvmlDeviceGetHandleByIndex(0))

However we would like to move to nvidia-ml-py in the future. Raising this issue to track this work

Metadata

Metadata

Assignees

Labels

build/testBuilding and testing the documentationimprovementImproves an existing functionality

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions