From ad07919f9220668ea68759bf08f4e323e01123fe Mon Sep 17 00:00:00 2001 From: Ben Hearsum Date: Mon, 20 Jan 2025 13:44:19 -0500 Subject: [PATCH] install nvidia driver in train image --- taskcluster/docker/train/Dockerfile | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/taskcluster/docker/train/Dockerfile b/taskcluster/docker/train/Dockerfile index 4768e1b33..f218e2a89 100644 --- a/taskcluster/docker/train/Dockerfile +++ b/taskcluster/docker/train/Dockerfile @@ -18,5 +18,13 @@ RUN apt-get update -qq \ libicu-dev \ && apt-get clean + +ENV DEBIAN_FRONTEND=noninteractive +RUN curl -sSO https://developer.download.nvidia.com/compute/nvidia-driver/565.57.01/local_installers/nvidia-driver-local-repo-ubuntu2204-565.57.01_1.0-1_amd64.deb \ + && dpkg -i nvidia-driver-local-repo-ubuntu2204-565.57.01_1.0-1_amd64.deb \ + && cp /var/nvidia-driver-local-repo-ubuntu2204-565.57.01/nvidia-driver-*-keyring.gpg /usr/share/keyrings/ \ + && apt-get update -qq \ + && apt-get install -y cuda-drivers-565 + VOLUME /builds/worker/checkouts VOLUME /builds/worker/.cache