Sign inSign up

determinedai/pytorch-ngc-hpc:51f8ef5

Manifest digest

sha256:e5a89f48ef0c3fb4e86839b7612706cb5557bf3483ea6759349235ab1362c2a4

OS/ARCH

linux/amd64

Compressed size

10.28 GB

Last pushed

over 2 years by detdockerci

Type

Image

Manifest digest

sha256:e5a89f48ef0c3fb4e86839b7612706cb5557bf3483ea6759349235ab1362c2a4

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.ref.name=ubuntu0 B
4LABEL org.opencontainers.image.version=22.040 B
5ADD file ... in / 29.04 MB
6CMD ["/bin/bash"]0 B
7RUN /bin/sh -c export DEBIAN_FRONTEND=noninteractive126.21 MB
8ARG CUDA_VERSION0 B
9ARG CUDA_DRIVER_VERSION0 B
10ARG JETPACK_HOST_MOUNTS0 B
11ENV CUDA_VERSION=12.3.2.001 CUDA_DRIVER_VERSION=545.23.08 CUDA_CACHE_DISABLE=1 NVIDIA_REQUIRE_JETPACK_HOST_MOUNTS=0 B
12RUN |3 CUDA_VERSION=12.3.2.001 CUDA_DRIVER_VERSION=545.23.08 JETPACK_HOST_MOUNTS=32 B
13RUN |3 CUDA_VERSION=12.3.2.001 CUDA_DRIVER_VERSION=545.23.08 JETPACK_HOST_MOUNTS=147.97 MB
14RUN |3 CUDA_VERSION=12.3.2.001 CUDA_DRIVER_VERSION=545.23.08 JETPACK_HOST_MOUNTS=14.73 KB
15ENV _CUDA_COMPAT_PATH=/usr/local/cuda/compat ENV=/etc/shinit_v2 BASH_ENV=/etc/bash.bashrc SHELL=/bin/bash0 B
16LABEL com.nvidia.volumes.needed=nvidia_driver com.nvidia.cuda.version=9.00 B
17ARG NCCL_VERSION0 B
18ARG CUBLAS_VERSION0 B
19ARG CUFFT_VERSION0 B
20ARG CURAND_VERSION0 B
21ARG CUSPARSE_VERSION0 B
22ARG CUSOLVER_VERSION0 B
23ARG CUTENSOR_VERSION0 B
24ARG NPP_VERSION0 B
25ARG NVJPEG_VERSION0 B
26ARG CUDNN_VERSION0 B
27ARG TRT_VERSION0 B
28ARG TRTOSS_VERSION0 B
29ARG NSIGHT_SYSTEMS_VERSION0 B
30ARG NSIGHT_COMPUTE_VERSION0 B
31ENV NCCL_VERSION=2.19.3 CUBLAS_VERSION=12.3.4.1 CUFFT_VERSION=11.0.12.1 CURAND_VERSION=10.3.4.1070 B
32RUN |17 CUDA_VERSION=12.3.2.001 CUDA_DRIVER_VERSION=545.23.08 JETPACK_HOST_MOUNTS=2.75 GB
33LABEL com.nvidia.nccl.version=2.19.3 com.nvidia.cublas.version=12.3.4.1 com.nvidia.cufft.version=11.0.12.1 com.nvidia.curand.version=10.3.4.1070 B
34ARG DALI_VERSION0 B
35ARG DALI_BUILD0 B
36ARG POLYGRAPHY_VERSION0 B
37ARG TRANSFORMER_ENGINE_VERSION0 B
38ENV DALI_VERSION=1.32.0 DALI_BUILD=10610166 POLYGRAPHY_VERSION=0.49.1 TRANSFORMER_ENGINE_VERSION=1.10 B
39ADD docs.tgz / # buildkit11.26 KB
40RUN |21 CUDA_VERSION=12.3.2.001 CUDA_DRIVER_VERSION=545.23.08 JETPACK_HOST_MOUNTS=179 B
41ARG _LIBPATH_SUFFIX0 B
42ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/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 B
43COPY entrypoint/ /opt/nvidia/ # buildkit5.59 KB
44ENV NVIDIA_PRODUCT_NAME=CUDA0 B
45ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]0 B
46COPY NVIDIA_Deep_Learning_Container_License.pdf /workspace/ # buildkit136.03 KB
47RUN /bin/sh -c export DEBIAN_FRONTEND=noninteractive29.9 MB
48ARG GDRCOPY_VERSION0 B
49ARG HPCX_VERSION0 B
50ARG RDMACORE_VERSION0 B
51ARG MOFED_VERSION=5.4-rdmacore39.00 B
52ARG OPENUCX_VERSION0 B
53ARG OPENMPI_VERSION0 B
54ENV GDRCOPY_VERSION=2.3 HPCX_VERSION=2.16rc4 MOFED_VERSION=5.4-rdmacore39.0 OPENUCX_VERSION=1.15.00 B
55ARG TARGETARCH0 B
56RUN |7 GDRCOPY_VERSION=2.3 HPCX_VERSION=2.16rc4 RDMACORE_VERSION=39.082.88 MB
57ENV OPAL_PREFIX=/opt/hpcx/ompi PATH=/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin0 B
58ENV OMPI_MCA_coll_hcoll_enable=00 B
59COPY cuda-*.patch /tmp # buildkit503 B
60RUN |7 GDRCOPY_VERSION=2.3 HPCX_VERSION=2.16rc4 RDMACORE_VERSION=39.0454.32 MB
61ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs:0 B
62ENV NVIDIA_PRODUCT_NAME=PyTorch0 B
63ARG NVIDIA_PYTORCH_VERSION0 B
64ARG PYTORCH_BUILD_VERSION0 B
65ENV PYTORCH_BUILD_VERSION=2.2.0a0+81ea7a4 PYTORCH_VERSION=2.2.0a0+81ea7a4 PYTORCH_BUILD_NUMBER=0 NVIDIA_PYTORCH_VERSION=23.120 B
66LABEL com.nvidia.pytorch.version=2.2.0a0+81ea7a40 B
67ARG TARGETARCH0 B
68ARG PYVER=3.100 B
69ARG L4T=00 B
70RUN |5 NVIDIA_PYTORCH_VERSION=23.12 PYTORCH_BUILD_VERSION=2.2.0a0+81ea7a4 TARGETARCH=amd6461.57 MB
71ENV PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python0 B
72RUN |5 NVIDIA_PYTORCH_VERSION=23.12 PYTORCH_BUILD_VERSION=2.2.0a0+81ea7a4 TARGETARCH=amd648.14 MB
73RUN |5 NVIDIA_PYTORCH_VERSION=23.12 PYTORCH_BUILD_VERSION=2.2.0a0+81ea7a4 TARGETARCH=amd6425.73 MB
74RUN |5 NVIDIA_PYTORCH_VERSION=23.12 PYTORCH_BUILD_VERSION=2.2.0a0+81ea7a4 TARGETARCH=amd6412.16 MB
75WORKDIR /opt/pytorch116 B
76COPY . . # buildkit476.19 MB
77ENV PYTHONIOENCODING=utf-80 B
78ENV LC_ALL=C.UTF-80 B
79ENV PIP_DEFAULT_TIMEOUT=1000 B
80RUN |5 NVIDIA_PYTORCH_VERSION=23.12 PYTORCH_BUILD_VERSION=2.2.0a0+81ea7a4 TARGETARCH=amd64611.72 MB
81RUN |5 NVIDIA_PYTORCH_VERSION=23.12 PYTORCH_BUILD_VERSION=2.2.0a0+81ea7a4 TARGETARCH=amd6452.3 MB
82RUN |5 NVIDIA_PYTORCH_VERSION=23.12 PYTORCH_BUILD_VERSION=2.2.0a0+81ea7a4 TARGETARCH=amd648.22 KB
83ENV NVM_DIR=/usr/local/nvm0 B
84RUN |5 NVIDIA_PYTORCH_VERSION=23.12 PYTORCH_BUILD_VERSION=2.2.0a0+81ea7a4 TARGETARCH=amd6466.03 MB
85COPY jupyter_notebook_config.py /usr/local/etc/jupyter/ # buildkit468 B
86ENV JUPYTER_PORT=88880 B
87ENV TENSORBOARD_PORT=60060 B
88EXPOSE map[8888/tcp:{}]0 B
89EXPOSE map[6006/tcp:{}]0 B
90RUN |5 NVIDIA_PYTORCH_VERSION=23.12 PYTORCH_BUILD_VERSION=2.2.0a0+81ea7a4 TARGETARCH=amd6415.02 MB
91ENV UCC_CL_BASIC_TLS=^sharp0 B
92ENV TORCH_CUDA_ARCH_LIST=5.2 6.0 6.1 7.00 B
93ENV PYTORCH_HOME=/opt/pytorch/pytorch0 B
94ENV CUDA_HOME=/usr/local/cuda0 B
95ENV TORCH_ALLOW_TF32_CUBLAS_OVERRIDE=10 B
96ENV USE_EXPERIMENTAL_CUDNN_V8_API=10 B
97RUN |5 NVIDIA_PYTORCH_VERSION=23.12 PYTORCH_BUILD_VERSION=2.2.0a0+81ea7a4 TARGETARCH=amd642.13 GB
98RUN |5 NVIDIA_PYTORCH_VERSION=23.12 PYTORCH_BUILD_VERSION=2.2.0a0+81ea7a4 TARGETARCH=amd64312.9 KB
99RUN /bin/sh -c pip install106.77 MB
100RUN /bin/sh -c if [280.87 MB
101ENV COCOAPI_VERSION=2.0+nv0.8.00 B
102RUN /bin/sh -c export COCOAPI_TAG=$(echo26.57 MB
103COPY singularity/ /.singularity.d/ # buildkit1.09 KB
104RUN /bin/sh -c ( cd447.15 MB
105RUN /bin/sh -c pip uninstall1.28 MB
106RUN /bin/sh -c pip install60.24 KB
107RUN /bin/sh -c if [1.66 GB
108WORKDIR /workspace32 B
109COPY NVREADME.md README.md # buildkit1.1 KB
110COPY docker-examples docker-examples # buildkit1.09 KB
111COPY examples examples # buildkit6.45 MB
112COPY tutorials tutorials # buildkit28.73 MB
113RUN /bin/sh -c chmod -R35.32 MB
114RUN /bin/sh -c set -x1.49 MB
115ENV PATH=/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin0 B
116RUN /bin/sh -c pip --version3.53 MB
117COPY torch_tensorrt/ /opt/pytorch/torch_tensorrt/ # buildkit57.68 MB
118ARG PYVER0 B
119RUN |1 PYVER=3.10 /bin/sh -c13.87 MB
120ENV LD_LIBRARY_PATH=/usr/local/lib/python3.10/dist-packages/torch/lib:/usr/local/lib/python3.10/dist-packages/torch_tensorrt/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib640 B
121ENV PATH=/usr/local/lib/python3.10/dist-packages/torch_tensorrt/bin:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin0 B
122RUN |1 PYVER=3.10 /bin/sh -c116.11 MB
123RUN |1 PYVER=3.10 /bin/sh -c104.14 MB
124ENV TORCH_CUDNN_V8_API_ENABLED=10 B
125ENV CUDA_MODULE_LOADING=LAZY0 B
126RUN |1 PYVER=3.10 /bin/sh -c13.76 KB
127COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit503 B
128ARG NVIDIA_BUILD_ID0 B
129ENV NVIDIA_BUILD_ID=764380080 B
130LABEL com.nvidia.build.id=764380080 B
131ARG NVIDIA_BUILD_REF0 B
132LABEL com.nvidia.build.ref=b3f725706edbd7cc21a22384237d38b2577047960 B
133/bin/sh -c chown root:root /usr/lib16.33 KB
134COPY dir:8d425e9faf0fbf76b0dff34795633c1e015b84964e7efa818ff34ffa07bb4b96 in /tmp/det_dockerfile_scripts 18.17 KB
135/bin/sh -c /tmp/det_dockerfile_scripts/install_deb_packages.sh74.67 MB
136/bin/sh -c python -m pip33.74 MB
137/bin/sh -c python -m pip38.17 MB
138/bin/sh -c python -m pip16.92 KB
139ENV JUPYTER_CONFIG_DIR=/run/determined/jupyter/config0 B
140ENV JUPYTER_DATA_DIR=/run/determined/jupyter/data0 B
141ENV JUPYTER_RUNTIME_DIR=/run/determined/jupyter/runtime0 B
142/bin/sh -c /tmp/det_dockerfile_scripts/add_det_nobody_user.sh41.31 KB
143/bin/sh -c /tmp/det_dockerfile_scripts/install_google_cloud_sdk.sh226.85 MB
144ARG DEEPSPEED_PIP0 B
145|1 DEEPSPEED_PIP=deepspeed==0.13.0 /bin/sh -c if27.8 MB
146|1 DEEPSPEED_PIP=deepspeed==0.13.0 /bin/sh -c /tmp/det_dockerfile_scripts/install_libnss_determined.sh26.05 KB
147|1 DEEPSPEED_PIP=deepspeed==0.13.0 /bin/sh -c chown17.07 KB
148|1 DEEPSPEED_PIP=deepspeed==0.13.0 /bin/sh -c rm17.5 KB
149COPY dir:8d425e9faf0fbf76b0dff34795633c1e015b84964e7efa818ff34ffa07bb4b96 in /tmp/det_dockerfile_scripts 18.17 KB
150ARG AWS_PLUGIN_INSTALL_DIR=/container/aws0 B
151ARG WITH_AWS_TRACE0 B
152ARG INTERNAL_AWS_DS0 B
153ARG INTERNAL_AWS_PATH0 B
154|1 AWS_PLUGIN_INSTALL_DIR=/container/aws /bin/sh -c if17.03 KB
155ARG AWS_LIB_DIR=/container/aws/lib0 B
156ENV LD_LIBRARY_PATH=/usr/local/lib/python3.10/dist-packages/torch/lib:/usr/local/lib/python3.10/dist-packages/torch_tensorrt/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib640 B
157|2 AWS_LIB_DIR=/container/aws/lib AWS_PLUGIN_INSTALL_DIR=/container/aws /bin/sh -c19.01 KB
158ENTRYPOINT ["/container/bin/scrape_libs.sh"]0 B
159|2 AWS_LIB_DIR=/container/aws/lib AWS_PLUGIN_INSTALL_DIR=/container/aws /bin/sh -c17.12 KB

Command

 ARG RELEASE