Sign inSign up

ytenzen/mxnet:tune-23.09

Index digest

sha256:1cf7af123b736bd41815d222fd0b56795d440e9a0dc59371b30cb1fe25c47c9e

OS/ARCH

linux/arm64

Compressed size

5.65 GB

Last pushed

almost 3 years by ytenzen

Type

Image

Manifest digest

sha256:7321a361760a7f41ef4d8b58fc151f39c3b2f9e68fbde8fabeb21045120c7f67

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 / 26.08 MB
6CMD ["/bin/bash"]0 B
7RUN /bin/sh -c export DEBIAN_FRONTEND=noninteractive118.82 MB
8ARG CUDA_VERSION0 B
9ARG CUDA_DRIVER_VERSION0 B
10ARG JETPACK_HOST_MOUNTS0 B
11ENV CUDA_VERSION=12.2.1.020 CUDA_DRIVER_VERSION=535.86.10 CUDA_CACHE_DISABLE=1 NVIDIA_REQUIRE_JETPACK_HOST_MOUNTS=0 B
12RUN |3 CUDA_VERSION=12.2.1.020 CUDA_DRIVER_VERSION=535.86.10 JETPACK_HOST_MOUNTS=138.39 MB
13RUN |3 CUDA_VERSION=12.2.1.020 CUDA_DRIVER_VERSION=535.86.10 JETPACK_HOST_MOUNTS=14.49 KB
14ENV _CUDA_COMPAT_PATH=/usr/local/cuda/compat ENV=/etc/shinit_v2 BASH_ENV=/etc/bash.bashrc SHELL=/bin/bash0 B
15LABEL com.nvidia.volumes.needed=nvidia_driver com.nvidia.cuda.version=9.00 B
16ARG NCCL_VERSION0 B
17ARG CUBLAS_VERSION0 B
18ARG CUFFT_VERSION0 B
19ARG CURAND_VERSION0 B
20ARG CUSPARSE_VERSION0 B
21ARG CUSOLVER_VERSION0 B
22ARG CUTENSOR_VERSION0 B
23ARG NPP_VERSION0 B
24ARG NVJPEG_VERSION0 B
25ARG CUDNN_VERSION0 B
26ARG TRT_VERSION0 B
27ARG TRTOSS_VERSION0 B
28ARG NSIGHT_SYSTEMS_VERSION0 B
29ARG NSIGHT_COMPUTE_VERSION0 B
30ENV NCCL_VERSION=2.18.5 CUBLAS_VERSION=12.2.5.6 CUFFT_VERSION=11.0.8.91 CURAND_VERSION=10.3.3.1290 B
31RUN |17 CUDA_VERSION=12.2.1.020 CUDA_DRIVER_VERSION=535.86.10 JETPACK_HOST_MOUNTS=2.75 GB
32LABEL com.nvidia.nccl.version=2.18.5 com.nvidia.cublas.version=12.2.5.6 com.nvidia.cufft.version=11.0.8.91 com.nvidia.curand.version=10.3.3.1290 B
33ARG DALI_VERSION0 B
34ARG DALI_BUILD0 B
35ARG POLYGRAPHY_VERSION0 B
36ARG TRANSFORMER_ENGINE_VERSION0 B
37ENV DALI_VERSION=1.29.0 DALI_BUILD=9289093 POLYGRAPHY_VERSION=0.49.0 TRANSFORMER_ENGINE_VERSION=0.120 B
38ADD docs.tgz / # buildkit11.26 KB
39RUN |21 CUDA_VERSION=12.2.1.020 CUDA_DRIVER_VERSION=535.86.10 JETPACK_HOST_MOUNTS=181 B
40ARG _LIBPATH_SUFFIX0 B
41ENV 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
42COPY entrypoint/ /opt/nvidia/ # buildkit5.58 KB
43ENV NVIDIA_PRODUCT_NAME=CUDA0 B
44ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]0 B
45COPY NVIDIA_Deep_Learning_Container_License.pdf /workspace/ # buildkit136.03 KB
46RUN /bin/sh -c export DEBIAN_FRONTEND=noninteractive29.78 MB
47ARG GDRCOPY_VERSION0 B
48ARG HPCX_VERSION0 B
49ARG RDMACORE_VERSION0 B
50ARG MOFED_VERSION=5.4-rdmacore39.00 B
51ARG OPENUCX_VERSION0 B
52ARG OPENMPI_VERSION0 B
53ENV GDRCOPY_VERSION=2.3 HPCX_VERSION=2.16rc4 MOFED_VERSION=5.4-rdmacore39.0 OPENUCX_VERSION=1.15.00 B
54ARG TARGETARCH0 B
55RUN |7 GDRCOPY_VERSION=2.3 HPCX_VERSION=2.16rc4 RDMACORE_VERSION=39.081.25 MB
56ENV 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
57ENV OMPI_MCA_coll_hcoll_enable=00 B
58COPY cuda-*.patch /tmp # buildkit503 B
59RUN |7 GDRCOPY_VERSION=2.3 HPCX_VERSION=2.16rc4 RDMACORE_VERSION=39.0305.23 MB
60ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs:0 B
61ENV NVIDIA_PRODUCT_NAME=MXNet0 B
62ARG MXNET_VERSION0 B
63ENV MXNET_VERSION=1.9.10 B
64LABEL com.nvidia.mxnet.version=1.9.10 B
65ARG NVIDIA_MXNET_VERSION0 B
66ENV NVIDIA_MXNET_VERSION=23.090 B
67ARG TARGETARCH0 B
68ARG PYVER=3.100 B
69ENV ONNX_NAMESPACE=onnx0 B
70RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm6457.08 MB
71ENV PYTHONIOENCODING=utf-80 B
72ENV LC_ALL=C.UTF-80 B
73RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm6413.5 MB
74RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm64152 B
75RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm64152 B
76RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm648.05 MB
77RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm6456.7 MB
78ENV SETUPTOOLS_USE_DISTUTILS=stdlib0 B
79RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm6446.93 MB
80RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm641.1 MB
81RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm647.8 MB
82RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm6459.77 MB
83RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm641.06 MB
84RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm6467.29 MB
85WORKDIR /opt/mxnet0 B
86COPY . . # buildkit179.54 MB
87ENV MXNET_HOME=/opt/mxnet0 B
88ENV MXNET_CUDNN_AUTOTUNE_DEFAULT=20 B
89ENV NCCL_DISABLE_CHECKS=10 B
90ENV MXNET_USE_OPERATOR_TUNING=00 B
91ENV MXNET_GPU_COPY_NTHREADS=10 B
92ENV MXNET_OPTIMIZER_AGGREGATION_SIZE=600 B
93ENV CUDA_DEVICE_MAX_CONNECTIONS=320 B
94ENV CUDA_MODULE_LOADING=LAZY0 B
95ENV LD_LIBRARY_PATH=/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/lib0 B
96RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm644.87 MB
97RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm644.21 MB
98RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm64155.08 MB
99RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm6432 B
100ENV NVSHMEM_BOOTSTRAP_PLUGIN=/usr/local/nvshmem/lib/nvshmem_bootstrap_mpi.so0 B
101RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm641.42 GB
102RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm64839.08 KB
103RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm6428.37 MB
104RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm6456.46 MB
105RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm644.69 KB
106RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm6450.09 MB
107RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm6496 B
108ENV 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
109WORKDIR /workspace0 B
110COPY NVREADME.md README.md # buildkit1.27 KB
111COPY docker-examples docker-examples # buildkit1002 B
112COPY nvidia-examples examples # buildkit8.84 MB
113COPY example/image-classification examples/image-classification # buildkit84.31 KB
114RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm649.06 MB
115RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm64181 B
116RUN |4 MXNET_VERSION=1.9.1 NVIDIA_MXNET_VERSION=23.09 TARGETARCH=arm6410.29 KB
117COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit352 B
118ARG NVIDIA_BUILD_ID0 B
119ENV NVIDIA_BUILD_ID=690848010 B
120LABEL com.nvidia.build.id=690848010 B
121ARG NVIDIA_BUILD_REF0 B
122LABEL com.nvidia.build.ref=9f9cb6adcef19dca956032db6b0c831d692356360 B
123COPY start-job.py / # buildkit579 B
124COPY auto-tuning.py / # buildkit2.52 KB

Command

 ARG RELEASE