Skip to content

Commit 79d8057

Browse files
authored
Update error comment
1 parent ac62305 commit 79d8057

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

codecarbon/core/gpu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def _get_memory_info(self):
144144
try:
145145
return pynvml.nvmlDeviceGetMemoryInfo(self.handle)
146146
except pynvml.NVMLError_NotSupported:
147-
# this error is currently thrown for the NVIDIA Blackwell GPU, due to memory sharing on NVIDIA DGX Spark
147+
# error thrown for the NVIDIA Blackwell GPU of DGX Spark, due to memory sharing -> return defaults instead
148148
return pynvml.c_nvmlMemory_t(-1, -1, -1)
149149

150150
def _get_temperature(self) -> int:

0 commit comments

Comments
 (0)