Merge pull request #611 from google:yiinho-nvidia-devtools-trust

PiperOrigin-RevId: 627085448
AI-Hypercomputer · Apr 22, 2024 · 49be620 · 49be620
2 parents c0bef1c + 5a3d5bd
commit 49be620
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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