torch-gpu (latest)

Published 2025-04-11 20:09:56 +00:00 by martin

Installation

docker pull gitea.hnatekmar.xyz/public/torch-gpu:latest
sha256:28635a2882989cf3ce3b6049ab72b03d383825643f1225f85b4b587322e07f8b

Image Layers

ARG RELEASE
ARG LAUNCHPAD_BUILD_ARCH
LABEL org.opencontainers.image.ref.name=ubuntu
LABEL org.opencontainers.image.version=22.04
ADD file:63d5ab3ef0aab308c0e71cb67292c5467f60deafa9b0418cbb220affcd078444 in /
CMD ["/bin/bash"]
ENV NVARCH=x86_64
ENV NVIDIA_REQUIRE_CUDA=cuda>=12.3 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=titan,driver>=470,driver<471 brand=titanrtx,driver>=470,driver<471 brand=tesla,driver>=525,driver<526 brand=unknown,driver>=525,driver<526 brand=nvidia,driver>=525,driver<526 brand=nvidiartx,driver>=525,driver<526 brand=geforce,driver>=525,driver<526 brand=geforcertx,driver>=525,driver<526 brand=quadro,driver>=525,driver<526 brand=quadrortx,driver>=525,driver<526 brand=titan,driver>=525,driver<526 brand=titanrtx,driver>=525,driver<526 brand=tesla,driver>=535,driver<536 brand=unknown,driver>=535,driver<536 brand=nvidia,driver>=535,driver<536 brand=nvidiartx,driver>=535,driver<536 brand=geforce,driver>=535,driver<536 brand=geforcertx,driver>=535,driver<536 brand=quadro,driver>=535,driver<536 brand=quadrortx,driver>=535,driver<536 brand=titan,driver>=535,driver<536 brand=titanrtx,driver>=535,driver<536
ENV NV_CUDA_CUDART_VERSION=12.3.52-1
ENV NV_CUDA_COMPAT_PACKAGE=cuda-compat-12-3
ARG TARGETARCH
LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>
RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends gnupg2 curl ca-certificates && curl -fsSLO https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/${NVARCH}/cuda-keyring_1.0-1_all.deb && dpkg -i cuda-keyring_1.0-1_all.deb && apt-get purge --autoremove -y curl && rm -rf /var/lib/apt/lists/* # buildkit
ENV CUDA_VERSION=12.3.0
RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-12-3=${NV_CUDA_CUDART_VERSION} ${NV_CUDA_COMPAT_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit
RUN |1 TARGETARCH=amd64 /bin/sh -c echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf # buildkit
ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64
COPY NGC-DL-CONTAINER-LICENSE / # buildkit
ENV NVIDIA_VISIBLE_DEVICES=all
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
ENV DEBIAN_FRONTEND=noninteractive
ENV LANG=C.UTF-8
COPY setup.sources.sh /setup.sources.sh # buildkit
COPY setup.packages.sh /setup.packages.sh # buildkit
COPY gpu.packages.txt /gpu.packages.txt # buildkit
RUN /bin/sh -c /setup.sources.sh # buildkit
RUN /bin/sh -c /setup.packages.sh /gpu.packages.txt # buildkit
ARG PYTHON_VERSION=python3.11
ARG TENSORFLOW_PACKAGE=tensorflow==2.19.0
COPY setup.python.sh /setup.python.sh # buildkit
COPY gpu.requirements.txt /gpu.requirements.txt # buildkit
RUN |2 PYTHON_VERSION=python3.11 TENSORFLOW_PACKAGE=tensorflow==2.19.0 /bin/sh -c /setup.python.sh $PYTHON_VERSION /gpu.requirements.txt # buildkit
RUN |2 PYTHON_VERSION=python3.11 TENSORFLOW_PACKAGE=tensorflow==2.19.0 /bin/sh -c pip install --no-cache-dir ${TENSORFLOW_PACKAGE} # buildkit
COPY setup.cuda.sh /setup.cuda.sh # buildkit
RUN |2 PYTHON_VERSION=python3.11 TENSORFLOW_PACKAGE=tensorflow==2.19.0 /bin/sh -c /setup.cuda.sh # buildkit
COPY bashrc /etc/bash.bashrc # buildkit
RUN |2 PYTHON_VERSION=python3.11 TENSORFLOW_PACKAGE=tensorflow==2.19.0 /bin/sh -c chmod a+rwx /etc/bash.bashrc # buildkit
/bin/sh -c apt-get update && apt-get install -y curl
CMD ["/bin/bash"]

Labels

Key Value
io.buildah.version 1.39.2
maintainer NVIDIA CORPORATION <cudatools@nvidia.com>
org.opencontainers.image.ref.name ubuntu
org.opencontainers.image.version 22.04
Details
Container
2025-04-11 20:09:56 +00:00
2
OCI / Docker
linux/amd64
3.9 GiB
Versions (1) View all
latest 2025-04-11