From 4439d3af69eb27c755f7f9751fae1eb48b9b9559 Mon Sep 17 00:00:00 2001
From: Simon Alibert <simon.alibert@huggingface.co>
Date: Tue, 10 Sep 2024 19:19:15 +0200
Subject: [PATCH] Try different base image

---
 docker/lerobot-gpu-dev/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/lerobot-gpu-dev/Dockerfile b/docker/lerobot-gpu-dev/Dockerfile
index d9ed8ca3e..93a38dad3 100644
--- a/docker/lerobot-gpu-dev/Dockerfile
+++ b/docker/lerobot-gpu-dev/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.2.2-devel-ubuntu22.04
+FROM nvidia/cuda:12.4.2-base-ubuntu22.04
 
 # Configure image
 ARG PYTHON_VERSION=3.10