From 2d991ce6f0c8cf47743cdca2c3c6f4d633c84067 Mon Sep 17 00:00:00 2001 From: Erik Sundell Date: Thu, 2 Sep 2021 02:39:22 +0200 Subject: [PATCH] Trial install of cupy and cudatoolkit --- hub.jupytearth.org-image/Dockerfile | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/hub.jupytearth.org-image/Dockerfile b/hub.jupytearth.org-image/Dockerfile index 8320664..c142c58 100644 --- a/hub.jupytearth.org-image/Dockerfile +++ b/hub.jupytearth.org-image/Dockerfile @@ -196,6 +196,23 @@ RUN echo "Installing conda packages..." \ google-cloud-sdk \ sympy \ # + # GPU related + # + # We work against a specific NVIDIA Driver, so we should install the + # latest cudatoolkit that supports that driver or update the driver to + # support the cudatoolkit we want to install. Since cudatoolkit may + # update and require more modern drivers than available, we pin this + # explicitly to avoid such issues. + # + # I think the driver will be updatable by editing a DaemonSet installed + # in the k8s cluster automatically by eksctl. + # + # If the wanted cudatoolkit version is know, then what driver should be + # used? Answer: https://docs.cupy.dev/en/stable/install.html#installation + # + cupy \ + cudatoolkit=11.4 \ + # # temp workarounds && echo "Installing conda packages complete!"