Add pynvml dependency (#237)

oliverholworthy · oliverholworthy · commit 6934be55f56e · 2023-03-10T09:30:27.000Z
* Add pynvml dependency

Correct name of pynvml package

* Add comment to requirements.txt about pynvml version

* Remove whitespace from requirements.txt
diff --git a/requirements.txt b/requirements.txt
@@ -9,3 +9,6 @@ tensorflow-metadata>=1.2.0
 betterproto<2.0.0
 packaging
 fsspec==2022.5.0
+
+# pynvml==11.5.0 is incompatible with distributed<2023.2.1
+pynvml>=11.0.0,<11.5
