From 308642502bac9425c80ebe7ddbd134bc6c281230 Mon Sep 17 00:00:00 2001 From: Carl Date: Tue, 1 Oct 2024 11:48:27 -0700 Subject: [PATCH] gpu issues? --- .github/workflows/docker-gpu-image.yml | 4 ++-- images/rl-env.yml | 5 +++-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/.github/workflows/docker-gpu-image.yml b/.github/workflows/docker-gpu-image.yml index 2ee7451..247c4e2 100644 --- 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 index 52a9be6..a097f83 100644 --- a/images/rl-env.yml +++ b/images/rl-env.yml @@ -8,6 +8,7 @@ dependencies: - huggingface_hub - nox - nvitop + - glances - plotnine - PyYaml - pytest @@ -15,6 +16,6 @@ dependencies: - sb3-contrib - scikit-optimize - pip: - - ray[rllib,tune] - - huggingface_sb3 +# - ray[rllib,tune] +# - huggingface_sb3