docker: include CUDA toolkit

SyneRBI · Jul 1, 2024 · 6066d9b · 6066d9b
1 parent 0520872
commit 6066d9b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/docker-compose.gpu.yml b/docker/docker-compose.gpu.yml
@@ -5,7 +5,7 @@ services:
   image: synerbi/jupyter:foundation-gpu
   build:
    args:
-    ROOT_CONTAINER: nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu22.04
+    ROOT_CONTAINER: nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
     PYTHON_VERSION: "3.10"
  base:
   image: synerbi/jupyter:base-gpu