diff --git a/taskcluster/docker/train/Dockerfile b/taskcluster/docker/train/Dockerfile
index 4768e1b33..caf557f34 100644
--- a/taskcluster/docker/train/Dockerfile
+++ b/taskcluster/docker/train/Dockerfile
@@ -18,5 +18,12 @@ RUN apt-get update -qq \
                           libicu-dev \
     && apt-get clean
 
+
+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 \
+    && apt-get install -y cuda-drivers-565
+
 VOLUME /builds/worker/checkouts
 VOLUME /builds/worker/.cache