From 330f4cf880596a32f6fb96ab287c5037a4f346b0 Mon Sep 17 00:00:00 2001 From: Ben Hearsum Date: Thu, 19 Dec 2024 11:23:58 -0500 Subject: [PATCH] install cuda --- taskcluster/docker/train/Dockerfile | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/taskcluster/docker/train/Dockerfile b/taskcluster/docker/train/Dockerfile index 719322066..fb1ea58bb 100644 --- a/taskcluster/docker/train/Dockerfile +++ b/taskcluster/docker/train/Dockerfile @@ -1,6 +1,10 @@ FROM $DOCKER_IMAGE_PARENT LABEL maintainer="Mozilla Release Engineering " +RUN curl -L https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb > /tmp/cuda-keyring.deb \ + && dpkg -i /tmp/cuda-keyring.deb \ + && rm /tmp/cuda-keyring.deb + RUN apt-get update -qq \ && apt-get install -y python3-numpy \ python3-fasttext \ @@ -16,6 +20,17 @@ RUN apt-get update -qq \ wget \ pkg-config \ libicu-dev \ + cuda-toolkit \ + software-properties-common \ + && apt-get clean + +RUN curl -L https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin > /tmp/cuda.pin \ + && mv /tmp/cuda.pin /etc/apt/preferences.d/cuda-repository-pin-600 \ + && apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub \ + && add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /" \ + && apt-get update -qq \ + && apt-get install -y cudnn9-cuda-12 \ + libcudnn9-dev-cuda-12 \ && apt-get clean