Update environment_gpu.yml

nasa-nccs-hpda · Jun 21, 2024 · 3eba08a · 3eba08a
1 parent 8b6fee0
commit 3eba08a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/requirements/environment_gpu.yml b/requirements/environment_gpu.yml
@@ -40,4 +40,6 @@ dependencies:
     - termcolor
     - numba
     - joblib
+    - timm
+    - deepspeed
     - segmentation-models-pytorch