gpu issues?

boettiger-lab · Oct 1, 2024 · 3086425 · 3086425
1 parent a173028
commit 3086425
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/.github/workflows/docker-gpu-image.yml b/.github/workflows/docker-gpu-image.yml
@@ -1,8 +1,8 @@
-name: Docker GPU Image
+name: GPU Image
 on:
   workflow_dispatch: null
   push:
-    paths: ['images/Dockerfile.gpu']
+    paths: ['images/Dockerfile.gpu', 'images/rl-env.yml', 'images/spatial-env.yml', 'images/jupyter-ai.yml']
 jobs:
   build:
     runs-on: ubuntu-latest

diff --git a/images/rl-env.yml b/images/rl-env.yml
@@ -8,13 +8,14 @@ dependencies:
   - huggingface_hub
   - nox
   - nvitop
+  - glances
   - plotnine
   - PyYaml
   - pytest
   - stable-baselines3
   - sb3-contrib
   - scikit-optimize
   - pip:
-    - ray[rllib,tune]
-    - huggingface_sb3
+#    - ray[rllib,tune]
+#    - huggingface_sb3