sha256:0b3cff2f89613747b2597f06a33aa0d73ef3905aed65500b4004ad31ff8548cf
OS/ARCH
Compressed size
9.88 GB
Last pushed
3 months by clustermax
Type
Image
Manifest digest
sha256:949329419e0e2d98b993c462a838aa2a913be43c6ce17f7802e97eb0d66fb4f7
Image Layers
1ARG RELEASE0 B2ARG LAUNCHPAD_BUILD_ARCH0 B3LABEL org.opencontainers.image.version=24.040 B4ADD file ... in / 28.36 MB5CMD ["/bin/bash"]0 B6ARG ENABLE_MITMPROXY=00 B7RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c32 B8RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c1 MB9RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c32 B10RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c3.43 MB11ARG GDRCOPY_VERSION=2.5.10 B12ARG HPCX_VERSION=2.260 B13ARG RDMACORE_VERSION=59.10 B14ARG MOFED_VERSION=5.4-rdmacore59.10 B15ARG OPENUCX_VERSION=1.20.00 B16ARG OPENMPI_VERSION=4.1.70 B17ARG EFA_VERSION=1.46.00 B18ARG AWS_OFI_NCCL_VERSION=1.17.30 B19ENV GDRCOPY_VERSION=2.5.1 HPCX_VERSION=2.26 MOFED_VERSION=5.4-rdmacore59.1 OPENUCX_VERSION=1.20.00 B20ARG TARGETARCH=amd640 B21RUN |10 ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.1 HPCX_VERSION=2.26196.53 MB22ENV OPAL_PREFIX=/usr/local/mpi PATH=/usr/local/mpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/amazon/efa/bin0 B23ENV OMPI_MCA_coll_hcoll_enable=00 B24ARG CUDA_VERSION=13.2.1.0090 B25ARG CUDA_DRIVER_VERSION=595.58.030 B26ARG NVVM_VERSION=13.2.780 B27ARG DOCA_VERSION=3.2.00 B28ENV CUDA_VERSION=13.2.1.009 CUDA_DRIVER_VERSION=595.58.03 NVVM_VERSION=13.2.78 DOCA_VERSION=3.2.00 B29RUN |14 ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.1 HPCX_VERSION=2.26154.59 MB30RUN |14 ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.1 HPCX_VERSION=2.2614.37 KB31ENV _CUDA_COMPAT_PATH=/usr/local/cuda/compat ENV=/etc/shinit_v2 BASH_ENV=/etc/bash.bashrc SHELL=/bin/bash0 B32LABEL com.nvidia.volumes.needed=nvidia_driver com.nvidia.cuda.version=9.00 B33ARG NCCL_VERSION=2.29.7 CUBLAS_VERSION=13.4.0.1 CUFFT_VERSION=12.2.0.46 CURAND_VERSION=10.4.2.550 B34ENV NCCL_VERSION=2.29.7 CUBLAS_VERSION=13.4.0.1 CUFFT_VERSION=12.2.0.46 CURAND_VERSION=10.4.2.550 B35ADD docs.tgz / # buildkit11.26 KB36RUN |47 ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.1 HPCX_VERSION=2.26179 B37ARG _LIBPATH_SUFFIX=0 B38ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/mpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/amazon/efa/bin LD_LIBRARY_PATH=/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64 NVIDIA_VISIBLE_DEVICES=all NVIDIA_DRIVER_CAPABILITIES=compute,utility,video0 B39RUN |48 ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.1 HPCX_VERSION=2.2692.38 MB40COPY /opt/amazon/aws-ofi-nccl /opt/amazon/aws-ofi-nccl # buildkit3.16 MB41RUN |48 ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.1 HPCX_VERSION=2.268.43 KB42COPY entrypoint/ /opt/nvidia/ # buildkit5.98 KB43ENV NVIDIA_PRODUCT_NAME=CUDA0 B44ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]0 B45RUN |48 ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.1 HPCX_VERSION=2.26408 B46RUN |48 ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.1 HPCX_VERSION=2.2632 B47ARG ENABLE_MITMPROXY=00 B48RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c32 B49RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c110.03 MB50ARG CUDA_COMPONENT_LIST=cccl crt nvrtc driver-dev0 B51ENV CUDA_COMPONENT_LIST=cccl crt nvrtc driver-dev0 B52RUN |2 ENABLE_MITMPROXY=0 CUDA_COMPONENT_LIST=cccl crt214.88 MB53ARG VERSION_LIST=CURAND_VERSION CUBLAS_VERSION CUFILE_VERSION0 B54RUN |3 ENABLE_MITMPROXY=0 CUDA_COMPONENT_LIST=cccl crt1.65 GB55ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs:0 B56RUN |3 ENABLE_MITMPROXY=0 CUDA_COMPONENT_LIST=cccl crt32 B57LABEL com.nvidia.nccl.version=2.29.7 com.nvidia.cublas.version=13.4.0.1 com.nvidia.cufft.version=12.2.0.46 com.nvidia.curand.version=10.4.2.550 B58ARG ENABLE_MITMPROXY=00 B59RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c32 B60RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c32 B61RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c64.13 MB62RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c2.51 GB63LABEL com.nvidia.nccl.version=2.29.7 com.nvidia.cublas.version=13.4.0.1 com.nvidia.cufft.version=12.2.0.46 com.nvidia.curand.version=10.4.2.550 B64RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c32 B65ARG TARGETARCH=amd640 B66ARG ENABLE_FIPS=00 B67ARG ENABLE_MITMPROXY=00 B68RUN |3 TARGETARCH=amd64 ENABLE_FIPS=0 ENABLE_MITMPROXY=032 B69RUN |3 TARGETARCH=amd64 ENABLE_FIPS=0 ENABLE_MITMPROXY=021.25 MB70COPY cuda-*.patch /tmp # buildkit503 B71RUN |3 TARGETARCH=amd64 ENABLE_FIPS=0 ENABLE_MITMPROXY=0345.2 MB72ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs:/usr/local/cuda/lib64/stubs:0 B73RUN |3 TARGETARCH=amd64 ENABLE_FIPS=0 ENABLE_MITMPROXY=032 B74ENV NVIDIA_PRODUCT_NAME=PyTorch0 B75ARG NVIDIA_PYTORCH_VERSION=26.040 B76ARG PYTORCH_BUILD_VERSION=2.12.0a0+0291f960 B77ARG NVFUSER_BUILD_VERSION=dd866a70 B78ENV PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 PYTORCH_VERSION=2.12.0a0+0291f96 PYTORCH_BUILD_NUMBER=0 NVIDIA_PYTORCH_VERSION=26.040 B79ENV NVFUSER_BUILD_VERSION=dd866a7 NVFUSER_VERSION=dd866a70 B80ARG TORCHAO_BUILD_VERSION=+git42bcdc490 B81ENV TORCHAO_BUILD_VERSION=+git42bcdc490 B82ARG TORCHTITAN_BUILD_VERSION=0.2.2+git6d1ff9e60 B83ENV TORCHTITAN_BUILD_VERSION=0.2.2+git6d1ff9e60 B84LABEL com.nvidia.pytorch.version=2.12.0a0+0291f960 B85ARG TARGETARCH=amd640 B86ARG PYVER=3.120 B87ENV PIP_BREAK_SYSTEM_PACKAGES=10 B88ARG ENABLE_MITMPROXY=00 B89RUN |8 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a732 B90RUN |8 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a7288.75 MB91ENV PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python0 B92ENV PIP_CONSTRAINT=/etc/pip/constraint.txt0 B93COPY constraint.txt /etc/pip/constraint.txt # buildkit372 B94RUN |8 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a731.85 MB95RUN |8 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a732 B96ENV NVPL_LAPACK_MATH_MODE=PEDANTIC0 B97WORKDIR /opt/pytorch115 B98COPY pytorch/requirements.txt pytorch/requirements-build.txt ./pytorch/ #585 B99ENV PYTHONIOENCODING=utf-80 B100ENV LC_ALL=C.UTF-80 B101ENV PIP_DEFAULT_TIMEOUT=1000 B102RUN |8 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a7727.83 MB103COPY . . # buildkit549.57 MB104RUN |8 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a781.88 MB105RUN |8 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a78.37 KB106WORKDIR /opt/pytorch32 B107RUN |8 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a75.15 MB108COPY jupyter_config/jupyter_notebook_config.py /usr/local/etc/jupyter/ # buildkit531 B109COPY jupyter_config/manager.jupyterlab-settings /root/.jupyter/lab/user-settings/@jupyterlab/completer-extension/ # buildkit376 B110COPY jupyter_config/settings.jupyterlab-settings /root/.jupyter/lab/user-settings/@jupyterlab/completer-extension/ # buildkit365 B111ENV JUPYTER_PORT=88880 B112ENV TENSORBOARD_PORT=60060 B113EXPOSE [8888/tcp]0 B114EXPOSE [6006/tcp]0 B115ENV UCC_CL_BASIC_TLS=^sharp0 B116ENV UCC_EC_CUDA_EXEC_NUM_THREADS=2560 B117ENV TORCH_CUDA_ARCH_LIST=7.5 8.0 8.6 9.00 B118ENV PYTORCH_HOME=/opt/pytorch/pytorch0 B119ENV CUDA_HOME=/usr/local/cuda0 B120ENV TORCH_ALLOW_TF32_CUBLAS_OVERRIDE=10 B121RUN |8 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a7623.61 MB122COPY /usr/local/share/cmake/TorchVision/ /usr/local/share/cmake/TorchVision/ # buildkit2.92 KB123COPY /usr/local/include/torchvision/ /usr/local/include/torchvision/ # buildkit71.38 KB124COPY /usr/local/lib64/libtorchvision.so.1.0 /usr/local/lib/libtorchvision.so.1.0 # buildkit2.23 MB125COPY /usr/local/lib64/libjpeg* /usr/local/lib/ # buildkit32 B126RUN |8 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a7130.34 KB127RUN |8 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a7300.86 MB128ENV TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas0 B129ENV TRITON_CUOBJDUMP_PATH=/usr/local/cuda/bin/cuobjdump0 B130ENV TRITON_NVDISASM_PATH=/usr/local/cuda/bin/nvdisasm0 B131ENV TRITON_CUDACRT_PATH=/usr/local/cuda/include0 B132ENV TRITON_CUDART_PATH=/usr/local/cuda/include0 B133ENV TRITON_CUPTI_LIB_PATH=/usr/local/cuda/lib640 B134ENV TRITON_CUPTI_INCLUDE_PATH=/usr/local/cuda/include0 B135ARG DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali0 B136RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a7240.22 MB137ENV COCOAPI_VERSION=2.0+nv0.8.10 B138RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a722.39 MB139COPY singularity/ /.singularity.d/ # buildkit1.09 KB140RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a732 B141RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a7254.99 MB142RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a780.67 KB143WORKDIR /workspace32 B144COPY NVREADME.md README.md # buildkit1.09 KB145COPY docker-examples docker-examples # buildkit1.09 KB146COPY tutorials tutorials # buildkit28.73 MB147RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a728.73 MB148RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a71.35 MB149ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/mpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/amazon/efa/bin:/opt/tensorrt/bin0 B150RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a720.48 MB151COPY torch_tensorrt/ /opt/pytorch/torch_tensorrt/ # buildkit159.77 MB152ARG PYVER=3.120 B153ENV LD_LIBRARY_PATH=/usr/local/lib/python3.12/dist-packages/torch/lib:/usr/local/lib/python3.12/dist-packages/torch_tensorrt/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib640 B154ENV PATH=/usr/local/lib/python3.12/dist-packages/torch_tensorrt/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/mpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/amazon/efa/bin:/opt/tensorrt/bin0 B155RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a7160.31 MB156RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a715.61 MB157RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a7444.68 MB158RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a7419.71 MB159RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a732 B160ENV CUDA_BINARY_LOADER_THREAD_COUNT=80 B161ENV CUDA_MODULE_LOADING=LAZY0 B162ENV TORCH_NCCL_USE_COMM_NONBLOCKING=00 B163ENV TORCHINDUCTOR_LOOP_ORDERING_AFTER_FUSION=00 B164RUN |9 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a718.12 KB165COPY restricted_constraint.txt /etc/pip/constraint.txt # buildkit153 B166COPY constraint.txt /etc/pip/original_constraint.txt # buildkit381 B167COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit501 B168ARG ENABLE_FIPS=00 B169RUN |10 NVIDIA_PYTORCH_VERSION=26.04 PYTORCH_BUILD_VERSION=2.12.0a0+0291f96 NVFUSER_BUILD_VERSION=dd866a732 B170ARG NVIDIA_BUILD_ID=2974449110 B171ENV NVIDIA_BUILD_ID=2974449110 B172LABEL com.nvidia.build.id=2974449110 B173ARG NVIDIA_BUILD_REF=be00fd5ad4299861d8ec6e69b73eb268da935bf20 B174LABEL com.nvidia.build.ref=be00fd5ad4299861d8ec6e69b73eb268da935bf20 B175ARG AV_VERSION=17.1.00 B176ARG PYARROW_VERSION=24.0.00 B177ARG TQDM_VERSION=4.68.20 B178LABEL org.opencontainers.image.title=cmax-datagen org.opencontainers.image.vendor=SemiAnalysis org.opencontainers.image.source=https://github.com/SemiAnalysisAI/ClusterMAX org.opencontainers.image.description=Pinned0 B179ENV PIP_DISABLE_PIP_VERSION_CHECK=1 PIP_NO_CACHE_DIR=1 PYTHONDONTWRITEBYTECODE=1 XDG_CACHE_HOME=/tmp/.cache0 B180RUN |3 AV_VERSION=17.1.0 PYARROW_VERSION=24.0.0 TQDM_VERSION=4.68.2134.22 MB181RUN |3 AV_VERSION=17.1.0 PYARROW_VERSION=24.0.0 TQDM_VERSION=4.68.283.39 MB182RUN |3 AV_VERSION=17.1.0 PYARROW_VERSION=24.0.0 TQDM_VERSION=4.68.232 B183CMD ["python3" "-c" "import av,0 BCommand
ARG RELEASE