Sign inSign up

davidnet/flax-cnn-model:1.0.0

Manifest digest

sha256:232059539d0dbea3b2bd07ec1fa462fa6901e09226a0b40df5a90f7acf3c6c58

OS/ARCH

linux/amd64

Compressed size

4.5 GB

Last pushed

about 2 years by davidnet

Type

Image

Manifest digest

sha256:232059539d0dbea3b2bd07ec1fa462fa6901e09226a0b40df5a90f7acf3c6c58

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 / 29.15 MB
6CMD ["/bin/bash"]0 B
7ENV NVARCH=x86_640 B
8ENV NVIDIA_REQUIRE_CUDA=cuda>=12.6 brand=unknown,driver>=470,driver<471 brand=grid,driver>=470,driver<471 brand=tesla,driver>=470,driver<4710 B
9ENV NV_CUDA_CUDART_VERSION=12.6.68-10 B
10ARG TARGETARCH0 B
11LABEL maintainer=NVIDIA CORPORATION <[email protected]>0 B
12RUN |1 TARGETARCH=amd64 /bin/sh -c6.62 MB
13ENV CUDA_VERSION=12.6.10 B
14RUN |1 TARGETARCH=amd64 /bin/sh -c54.66 MB
15RUN |1 TARGETARCH=amd64 /bin/sh -c187 B
16ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
17ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib640 B
18COPY NGC-DL-CONTAINER-LICENSE / # buildkit6.72 KB
19ENV NVIDIA_VISIBLE_DEVICES=all0 B
20ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility0 B
21ENV NV_CUDA_LIB_VERSION=12.6.1-10 B
22ENV NV_NVTX_VERSION=12.6.68-10 B
23ENV NV_LIBNPP_VERSION=12.3.1.54-10 B
24ENV NV_LIBNPP_PACKAGE=libnpp-12-6=12.3.1.54-10 B
25ENV NV_LIBCUSPARSE_VERSION=12.5.3.3-10 B
26ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-12-60 B
27ENV NV_LIBCUBLAS_VERSION=12.6.1.4-10 B
28ENV NV_LIBCUBLAS_PACKAGE=libcublas-12-6=12.6.1.4-10 B
29ENV NV_LIBNCCL_PACKAGE_NAME=libnccl20 B
30ENV NV_LIBNCCL_PACKAGE_VERSION=2.22.3-10 B
31ENV NCCL_VERSION=2.22.3-10 B
32ENV NV_LIBNCCL_PACKAGE=libnccl2=2.22.3-1+cuda12.60 B
33ARG TARGETARCH0 B
34LABEL maintainer=NVIDIA CORPORATION <[email protected]>0 B
35RUN |1 TARGETARCH=amd64 /bin/sh -c1.29 GB
36RUN |1 TARGETARCH=amd64 /bin/sh -c57.87 KB
37COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit1.64 KB
38COPY nvidia_entrypoint.sh /opt/nvidia/ # buildkit1.49 KB
39ENV NVIDIA_PRODUCT_NAME=CUDA0 B
40ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]0 B
41ENV NV_CUDNN_VERSION=9.3.0.75-10 B
42ENV NV_CUDNN_PACKAGE_NAME=libcudnn9-cuda-120 B
43ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12=9.3.0.75-10 B
44ARG TARGETARCH0 B
45LABEL maintainer=NVIDIA CORPORATION <[email protected]>0 B
46LABEL com.nvidia.cudnn.version=9.3.0.75-10 B
47RUN |1 TARGETARCH=amd64 /bin/sh -c555.39 MB
48COPY /uv /bin/uv # buildkit11.64 MB
49WORKDIR /usr/src/app131 B
50COPY pyproject.toml uv.lock /usr/src/app/ #32.16 KB
51RUN /bin/sh -c uv sync2.57 GB
52ENV PATH=/usr/src/app/.venv/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
53COPY train.py /usr/src/app/ # buildkit2.19 KB
54CMD ["python" "train.py" "--train-steps" "1000"0 B

Command

 ARG RELEASE