nemo_rl.utils.nvml

Contenu du Module

Nom	Description
`nvml_context`	Gestionnaire de contexte pour l’initialisation et l’arrêt de NVML.
`device_id_to_physical_device_id`	Convertir un ID de périphérique logique en un ID de périphérique physique en tenant compte de CUDA_VISIBLE_DEVICES.
`get_device_uuid`	Obtenir l’UUID d’un périphérique CUDA à l’aide de NVML.
`get_free_memory_bytes`	Obtenir la mémoire libre d’un périphérique CUDA en octets à l’aide de NVML.

nemo_rl.utils.nvml.nvml_context() -> typing.Generator[None, None, None]

Gestionnaire de contexte pour l’initialisation et l’arrêt de NVML.

Lève : RuntimeError : Si l’initialisation de NVML échoue

nemo_rl.utils.nvml.device_id_to_physical_device_id(device_id: int) -> int

Convertir un ID de périphérique logique en un ID de périphérique physique en tenant compte de CUDA_VISIBLE_DEVICES.

nemo_rl.utils.nvml.get_device_uuid(device_idx: int) -> str

Obtenir l’UUID d’un périphérique CUDA à l’aide de NVML.

nemo_rl.utils.nvml.get_free_memory_bytes(device_idx: int) -> float

Obtenir la mémoire libre d’un périphérique CUDA en octets à l’aide de NVML.