We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d2e1f02 commit 29f9e55Copy full SHA for 29f9e55
pytorch/training/docker/2.9/py3/cu130/Dockerfile.gpu
@@ -145,7 +145,7 @@ RUN pip install --no-cache-dir nvidia-mathdx
145
# Install TE using instructions from https://docs.nvidia.com/deeplearning/transformer-engine/user-guide/installation.html
146
RUN pip install --no-cache-dir git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TE_VERSION} --no-build-isolation
147
148
-RUN curl -o /license.txt https://aws-dlc-licenses.s3.amazonaws.com/pytorch-2.8/license.txt
+RUN curl -o /license.txt https://aws-dlc-licenses.s3.amazonaws.com/pytorch-2.9/license.txt
149
150
COPY start_cuda_compat.sh /usr/local/bin/start_cuda_compat.sh
151
RUN chmod +x /usr/local/bin/start_cuda_compat.sh
0 commit comments