Merge pull request #915 from SyneRBI/docker-nvcc

docker: include CUDA toolkit
SyneRBI · Jul 1, 2024 · d725d08 · d725d08
2 parents 0520872 + 6066d9b
commit d725d08
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