diff --git a/maxtext_gpu_dependencies.Dockerfile b/maxtext_gpu_dependencies.Dockerfile index 30d1b9e90..389dd234e 100644 --- a/maxtext_gpu_dependencies.Dockerfile +++ b/maxtext_gpu_dependencies.Dockerfile @@ -2,6 +2,9 @@ ARG BASEIMAGE=ghcr.io/nvidia/jax:base FROM $BASEIMAGE +# Stopgaps measure to circumvent gpg key setup issue. +RUN echo "deb [trusted=yes] https://developer.download.nvidia.com/devtools/repos/ubuntu2204/amd64/ /" > /etc/apt/sources.list.d/devtools-ubuntu2204-amd64.list + # Install dependencies for adjusting network rto RUN apt-get update && apt-get install -y iproute2 ethtool lsof