Merge branch 'master' into fix-gpu-tests-and-failing-metrics

pytorch · Dec 3, 2024 · 50c9aeb · 50c9aeb
2 parents e1fd667 + 9e2763e
commit 50c9aeb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -21,7 +21,7 @@ mlflow
 neptune-client>=0.16.17
 tensorboard
 torchvision
-pynvml
+pynvml<12  # pynvml module was removed in 12.X, is not developed or maintained. We should replace pynvml with something else.
 clearml
 scikit-image
 py-rouge