Fix dask-cuda dependency (#379)

dewan-c · web-flow · commit c396fafeb846 · 2023-02-17T16:33:34.000-05:00
diff --git a/requirements.txt b/requirements.txt
@@ -14,6 +14,7 @@ numpy==1.19.2
 pandas==1.2.4
 protobuf==3.20.1
 psutil==5.6.7  # sagemaker-containers requires psutil 5.6.7
+pynvml==11.4.1
 python-dateutil==2.8.1
 requests==2.25.1
 retrying==1.3.3
