Try nvidia/cudagl

huggingface · Apr 21, 2024 · f28006e · f28006e
1 parent 8ccd19d
commit f28006e
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/.github/workflows/build_docker_images.yml b/.github/workflows/build_docker_images.yml
@@ -3,9 +3,9 @@
 name: Build Docker images (scheduled)
 
 on:
-  # pull_request:  # TODO(aliberts): for testing, delete before merge
-  #   branches:
-  #     - main
+  pull_request:  # TODO(aliberts): for testing, delete before merge
+    branches:
+      - main
   workflow_dispatch:
   workflow_call:
   schedule:

diff --git a/.github/workflows/nightly-tests.yml b/.github/workflows/nightly-tests.yml
@@ -3,9 +3,9 @@
 name: Self-hosted runner with slow tests (scheduled)
 
 on:
-  pull_request:  # TODO(aliberts): for testing, delete before merge
-    branches:
-      - main
+  # pull_request:  # TODO(aliberts): for testing, delete before merge
+  #   branches:
+  #     - main
   workflow_dispatch:
   schedule:
     - cron: "0 2 * * *"

diff --git a/docker/lerobot-gpu/Dockerfile b/docker/lerobot-gpu/Dockerfile
@@ -8,7 +8,7 @@ RUN conda create -y -n lerobot python=${PYTHON_VERSION}
 ENV PATH=/opt/conda/envs/lerobot/bin:$PATH
 
 # Stage 2
-FROM nvidia/opengl:1.0-glvnd-devel-ubuntu22.04 AS build-image
+FROM nvidia/cudagl:11.3.0-devel-ubuntu20.04 AS build-image
 ARG DEBIAN_FRONTEND=noninteractive
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \