From 6066d9be6bc7ae30664728e6b1da8ae30388f578 Mon Sep 17 00:00:00 2001
From: Casper da Costa-Luis <casper.dcl@physics.org>
Date: Mon, 1 Jul 2024 21:57:46 +0100
Subject: [PATCH] docker: include CUDA toolkit

---
 docker/docker-compose.gpu.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/docker-compose.gpu.yml b/docker/docker-compose.gpu.yml
index b8304aac..cd37ebdf 100644
--- 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