Sign inSign up

zauberzeug/classification-trainer:0.5.0

Multi-platform
Index digest

sha256:aff1aaf0428153dcad6505f33ce223f9244e6de669b57fd61607c18fcbbbc468

OS/ARCH

Compressed size

9.39 GB

Last pushed

about 1 month by zauberzeug

Type

Image

Manifest digest

sha256:90ca9b1f5f907eb776fa6a3f8aa53811e84fe211dc82d723d9cae80d881dd1b8

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=24.040 B
5ADD file ... in / 27.53 MB
6CMD ["/bin/bash"]0 B
7ARG JETPACK_HOST_MOUNTS=0 B
8ARG ENABLE_MITMPROXY=00 B
9ENV NVIDIA_REQUIRE_JETPACK_HOST_MOUNTS=0 B
10RUN |2 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 /bin/sh32 B
11RUN |2 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 /bin/sh32 B
12RUN |2 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 /bin/sh1008 KB
13RUN |2 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 /bin/sh32 B
14RUN |2 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 /bin/sh3.5 MB
15ARG GDRCOPY_VERSION=2.5.10 B
16ARG HPCX_VERSION=2.25.1-RC20 B
17ARG RDMACORE_VERSION=56.00 B
18ARG MOFED_VERSION=5.4-rdmacore56.00 B
19ARG OPENUCX_VERSION=1.20.00 B
20ARG OPENMPI_VERSION=4.1.70 B
21ARG EFA_VERSION=1.43.10 B
22ARG AWS_OFI_NCCL_VERSION=1.17.00 B
23ENV GDRCOPY_VERSION=2.5.1 HPCX_VERSION=2.25.1-RC2 MOFED_VERSION=5.4-rdmacore56.0 OPENUCX_VERSION=1.20.00 B
24ARG TARGETARCH=arm640 B
25RUN |11 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.183.07 MB
26ENV OPAL_PREFIX=/opt/hpcx/ompi 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 B
27ENV OMPI_MCA_coll_hcoll_enable=00 B
28ARG CUDA_VERSION=13.1.1.0060 B
29ARG CUDA_DRIVER_VERSION=590.48.010 B
30ARG NVVM_VERSION=13.1.1150 B
31ARG DOCA_VERSION=3.1.00 B
32ENV CUDA_VERSION=13.1.1.006 CUDA_DRIVER_VERSION=590.48.01 NVVM_VERSION=13.1.115 DOCA_VERSION=3.1.00 B
33RUN |15 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.1149.97 MB
34RUN |15 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.115.2 KB
35ENV _CUDA_COMPAT_PATH=/usr/local/cuda/compat ENV=/etc/shinit_v2 BASH_ENV=/etc/bash.bashrc SHELL=/bin/bash0 B
36LABEL com.nvidia.volumes.needed=nvidia_driver com.nvidia.cuda.version=9.00 B
37ARG NCCL_VERSION=2.29.stable.20260109 CUBLAS_VERSION=13.2.1.1 CUFFT_VERSION=12.1.0.78 CURAND_VERSION=10.4.1.810 B
38ENV NCCL_VERSION=2.29.stable.20260109 CUBLAS_VERSION=13.2.1.1 CUFFT_VERSION=12.1.0.78 CURAND_VERSION=10.4.1.810 B
39ADD docs.tgz / # buildkit11.26 KB
40RUN |46 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.1187 B
41ARG _LIBPATH_SUFFIX=0 B
42ENV 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 B
43RUN |47 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.158.08 MB
44COPY /opt/amazon/aws-ofi-nccl /opt/amazon/aws-ofi-nccl # buildkit3.07 MB
45RUN |47 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.17.79 KB
46COPY entrypoint/ /opt/nvidia/ # buildkit6.21 KB
47ENV NVIDIA_PRODUCT_NAME=CUDA0 B
48ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]0 B
49RUN |47 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.1407 B
50RUN |47 JETPACK_HOST_MOUNTS= ENABLE_MITMPROXY=0 GDRCOPY_VERSION=2.5.132 B
51ARG ENABLE_MITMPROXY=00 B
52RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c32 B
53RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c105.6 MB
54ARG CUDA_COMPONENT_LIST=cccl crt nvrtc driver-dev0 B
55ENV CUDA_COMPONENT_LIST=cccl crt nvrtc driver-dev0 B
56RUN |2 ENABLE_MITMPROXY=0 CUDA_COMPONENT_LIST=cccl crt202.35 MB
57ARG VERSION_LIST=CURAND_VERSION CUBLAS_VERSION CUFILE_VERSION0 B
58RUN |3 ENABLE_MITMPROXY=0 CUDA_COMPONENT_LIST=cccl crt1.92 GB
59ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs:0 B
60RUN |3 ENABLE_MITMPROXY=0 CUDA_COMPONENT_LIST=cccl crt32 B
61LABEL com.nvidia.nccl.version=2.29.stable.20260109 com.nvidia.cublas.version=13.2.1.1 com.nvidia.cufft.version=12.1.0.78 com.nvidia.curand.version=10.4.1.810 B
62ARG ENABLE_MITMPROXY=00 B
63RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c32 B
64RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c32 B
65RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c57.62 MB
66RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c2.79 GB
67LABEL com.nvidia.nccl.version=2.29.stable.20260109 com.nvidia.cublas.version=13.2.1.1 com.nvidia.cufft.version=12.1.0.78 com.nvidia.curand.version=10.4.1.810 B
68RUN |1 ENABLE_MITMPROXY=0 /bin/sh -c32 B
69ARG TARGETARCH=arm640 B
70ARG ENABLE_FIPS=00 B
71ARG ENABLE_MITMPROXY=00 B
72RUN |3 TARGETARCH=arm64 ENABLE_FIPS=0 ENABLE_MITMPROXY=032 B
73RUN |3 TARGETARCH=arm64 ENABLE_FIPS=0 ENABLE_MITMPROXY=021.11 MB
74COPY cuda-*.patch /tmp # buildkit501 B
75RUN |3 TARGETARCH=arm64 ENABLE_FIPS=0 ENABLE_MITMPROXY=0342.37 MB
76ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs:/usr/local/cuda/lib64/stubs:0 B
77RUN |3 TARGETARCH=arm64 ENABLE_FIPS=0 ENABLE_MITMPROXY=032 B
78ENV NVIDIA_PRODUCT_NAME=PyTorch0 B
79ARG NVIDIA_PYTORCH_VERSION=26.020 B
80ARG PYTORCH_BUILD_VERSION=2.11.0a0+eb65b360 B
81ARG NVFUSER_BUILD_VERSION=cf170f40 B
82ENV PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 PYTORCH_VERSION=2.11.0a0+eb65b36 PYTORCH_BUILD_NUMBER=0 NVIDIA_PYTORCH_VERSION=26.020 B
83ENV NVFUSER_BUILD_VERSION=cf170f4 NVFUSER_VERSION=cf170f40 B
84ARG TORCHAO_BUILD_VERSION=+gita89eaab20 B
85ENV TORCHAO_BUILD_VERSION=+gita89eaab20 B
86ARG TORCHTITAN_BUILD_VERSION=0.2.1+git9f211ec10 B
87ENV TORCHTITAN_BUILD_VERSION=0.2.1+git9f211ec10 B
88LABEL com.nvidia.pytorch.version=2.11.0a0+eb65b360 B
89ARG TARGETARCH=arm640 B
90ARG PYVER=3.120 B
91ENV PIP_BREAK_SYSTEM_PACKAGES=10 B
92ARG L4T=00 B
93ARG ENABLE_MITMPROXY=00 B
94RUN |9 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f432 B
95RUN |9 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f4285.51 MB
96ENV PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python0 B
97ENV PIP_CONSTRAINT=/etc/pip/constraint.txt0 B
98COPY constraint.txt /etc/pip/constraint.txt # buildkit361 B
99RUN |9 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f431.4 MB
100RUN |9 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f414.84 MB
101ENV NVPL_LAPACK_MATH_MODE=PEDANTIC0 B
102WORKDIR /opt/pytorch116 B
103COPY pytorch/requirements.txt pytorch/requirements-build.txt ./pytorch/ #595 B
104ENV PYTHONIOENCODING=utf-80 B
105ENV LC_ALL=C.UTF-80 B
106ENV PIP_DEFAULT_TIMEOUT=1000 B
107RUN |9 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f4271.61 MB
108COPY . . # buildkit510.61 MB
109RUN |9 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f474.62 MB
110RUN |9 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f48.36 KB
111WORKDIR /opt/pytorch32 B
112RUN |9 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f43.63 MB
113COPY jupyter_config/jupyter_notebook_config.py /usr/local/etc/jupyter/ # buildkit523 B
114COPY jupyter_config/manager.jupyterlab-settings /root/.jupyter/lab/user-settings/@jupyterlab/completer-extension/ # buildkit379 B
115COPY jupyter_config/settings.jupyterlab-settings /root/.jupyter/lab/user-settings/@jupyterlab/completer-extension/ # buildkit365 B
116ENV JUPYTER_PORT=88880 B
117ENV TENSORBOARD_PORT=60060 B
118EXPOSE [8888/tcp]0 B
119EXPOSE [6006/tcp]0 B
120ENV UCC_CL_BASIC_TLS=^sharp0 B
121ENV UCC_EC_CUDA_EXEC_NUM_THREADS=2560 B
122ENV TORCH_CUDA_ARCH_LIST=8.0 8.6 9.0 10.00 B
123ENV PYTORCH_HOME=/opt/pytorch/pytorch0 B
124ENV CUDA_HOME=/usr/local/cuda0 B
125ENV TORCH_ALLOW_TF32_CUBLAS_OVERRIDE=10 B
126RUN |9 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f4646.54 MB
127COPY /usr/local/share/cmake/TorchVision/ /usr/local/share/cmake/TorchVision/ # buildkit2.91 KB
128COPY /usr/local/include/torchvision/ /usr/local/include/torchvision/ # buildkit70.64 KB
129COPY /usr/local/lib64/libtorchvision.so.1.0 /usr/local/lib/libtorchvision.so.1.0 # buildkit2.77 MB
130COPY /usr/local/lib64/libjpeg* /usr/local/lib/ # buildkit32 B
131RUN |9 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f4127.09 KB
132RUN |9 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f4281.09 MB
133ENV TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas0 B
134ENV TRITON_CUOBJDUMP_PATH=/usr/local/cuda/bin/cuobjdump0 B
135ENV TRITON_NVDISASM_PATH=/usr/local/cuda/bin/nvdisasm0 B
136ENV TRITON_CUDACRT_PATH=/usr/local/cuda/include0 B
137ENV TRITON_CUDART_PATH=/usr/local/cuda/include0 B
138ENV TRITON_CUPTI_LIB_PATH=/usr/local/cuda/lib640 B
139ENV TRITON_CUPTI_INCLUDE_PATH=/usr/local/cuda/include0 B
140ARG DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali0 B
141RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f4221.69 MB
142ENV COCOAPI_VERSION=2.0+nv0.8.10 B
143RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f422.5 MB
144COPY singularity/ /.singularity.d/ # buildkit1.09 KB
145RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f432 B
146RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f418.13 MB
147RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f4191.91 MB
148RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f471.93 KB
149WORKDIR /workspace32 B
150COPY NVREADME.md README.md # buildkit1.09 KB
151COPY docker-examples docker-examples # buildkit1.09 KB
152COPY tutorials tutorials # buildkit28.73 MB
153RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f428.73 MB
154RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f41.17 MB
155ENV 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 B
156RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f423.92 MB
157COPY torch_tensorrt/ /opt/pytorch/torch_tensorrt/ # buildkit64.46 MB
158ARG PYVER=3.120 B
159ENV 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 B
160ENV 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 B
161RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f4171.34 MB
162RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f415.12 MB
163RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f4438.87 MB
164RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f4256.92 MB
165RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f432 B
166RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f447.69 KB
167ENV CUDA_BINARY_LOADER_THREAD_COUNT=80 B
168ENV CUDA_MODULE_LOADING=LAZY0 B
169ENV TORCH_NCCL_USE_COMM_NONBLOCKING=00 B
170ENV TORCHINDUCTOR_LOOP_ORDERING_AFTER_FUSION=00 B
171RUN |10 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f415.8 KB
172COPY restricted_constraint.txt /etc/pip/constraint.txt # buildkit152 B
173COPY constraint.txt /etc/pip/original_constraint.txt # buildkit369 B
174COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit499 B
175ARG ENABLE_FIPS=00 B
176ARG TARGETARCH=arm640 B
177RUN |11 NVIDIA_PYTORCH_VERSION=26.02 PYTORCH_BUILD_VERSION=2.11.0a0+eb65b36 NVFUSER_BUILD_VERSION=cf170f432 B
178ARG NVIDIA_BUILD_ID=2694431000 B
179ENV NVIDIA_BUILD_ID=2694431000 B
180LABEL com.nvidia.build.id=2694431000 B
181ARG NVIDIA_BUILD_REF=fdfaeb27df34739d5d656e2d94755cd8f42fa80e0 B
182LABEL com.nvidia.build.ref=fdfaeb27df34739d5d656e2d94755cd8f42fa80e0 B
183ENV LD_LIBRARY_PATH=/opt/hpcx/ucx/lib:/opt/hpcx/ucc/lib:/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 B
184ENV DEBIAN_FRONTEND=noninteractive0 B
185ENV TZ=Etc/UTC0 B
186RUN /bin/sh -c apt-get update30.28 MB
187COPY /uv /usr/local/bin/uv # buildkit19.93 MB
188WORKDIR /uv_env94 B
189COPY pyproject.toml uv.lock ./ #42.2 KB
190RUN /bin/sh -c pip install32 B
191RUN /bin/sh -c uv venv9.72 KB
192RUN /bin/sh -c uv sync83.52 MB
193ENV PATH=/uv_env/.venv/bin:/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 B
194COPY . /app # buildkit58.38 KB
195RUN /bin/sh -c rm -f32 B
196WORKDIR /app32 B
197EXPOSE [80/tcp]0 B
198ENV TZ=Europe/Amsterdam0 B
199CMD ["python" "/app/main.py"]0 B

Command

 ARG RELEASE