| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.version=24.04 |
| ADD file:43d479b270bbaf47965cfc86b37f4c517bda83ddefda6f708f98c3b2b7d15396 in / |
| CMD ["/bin/bash"] |
| ARG PYTORCH_VERSION=2.14.1 |
| ARG TRITON_VERSION= |
| ARG TARGETPLATFORM=linux/amd64 |
| ARG CUDA_VERSION=13.2.2 |
| LABEL com.nvidia.volumes.needed=nvidia_driver |
| RUN |4 PYTORCH_VERSION=2.14.1 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=13.2.2 /bin/sh -c apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates libjpeg-dev libpng-dev python-is-python3 python3 python3-dev python3-pip && rm -rf /var/lib/apt/lists/* # buildkit |
| COPY /usr/local/lib/python3.12 /usr/local/lib/python3.12 # buildkit |
| COPY /usr/local/bin /usr/local/bin # buildkit |
| RUN |4 PYTORCH_VERSION=2.14.1 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=13.2.2 /bin/sh -c if test -n "${CUDA_VERSION}" -a "${TARGETPLATFORM}" != "linux/arm64"; then apt-get update -qq && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends gcc && rm -rf /var/lib/apt/lists/*; fi # buildkit |
| ENV NVIDIA_VISIBLE_DEVICES=all |
| ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility |
| ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 |
| ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV PYTORCH_VERSION=2.14.1 |
| WORKDIR /workspace |
| ARG CUDA_VERSION=13.2.2 |
| ARG BUILD_TYPE=dev |
| RUN |5 PYTORCH_VERSION=2.14.1 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=13.2.2 BUILD_TYPE=dev /bin/sh -c if [ "${BUILD_TYPE}" = "dev" ] && [ -n "${CUDA_VERSION}" ]; then apt-get update && apt-get install -y --no-install-recommends wget gnupg2 ca-certificates && NVARCH=$(uname -m | sed 's/x86_64/x86_64/' | sed 's/aarch64/sbsa/') && wget -qO - https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/${NVARCH}/3bf863cc.pub | apt-key add - && echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/${NVARCH} /" > /etc/apt/sources.list.d/cuda.list && CUDA_PKG_VERSION=$(echo ${CUDA_VERSION} | cut -d'.' -f1,2 | tr '.' '-') && apt-get update && apt-get install -y --no-install-recommends cuda-toolkit-${CUDA_PKG_VERSION} && apt-get clean && rm -rf /var/lib/apt/lists/* && echo "/usr/local/cuda/lib64" >> /etc/ld.so.conf.d/cuda.conf && ldconfig; fi # buildkit |
| ENV PATH=/usr/local/cuda/bin:/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/cuda/lib64:/usr/local/nvidia/lib:/usr/local/nvidia/lib64 |
| ENV CUDA_HOME=/usr/local/cuda |
| COPY /usr/local/lib/python3.12 /usr/local/lib/python3.12 # buildkit |
| COPY /usr/local/bin /usr/local/bin # buildkit |
| COPY /opt/pytorch /opt/pytorch # buildkit |
| USER root |
| SHELL [/bin/bash -c] |
| RUN /bin/bash -c apt-get update && apt-get install -y --no-install-recommends python3-pip sudo && apt-get clean && rm -rf /var/lib/apt/lists/* && echo "root:kobaken" | chpasswd && EXISTING_USER=$(getent passwd 1000 | cut -d: -f1) && EXISTING_GROUP=$(getent group 1000 | cut -d: -f1) && if [ -n "$EXISTING_USER" ]; then groupmod -n kbkn "$EXISTING_GROUP" || true; usermod -l kbkn -g kbkn -d /home/kbkn -m "$EXISTING_USER" || true; else groupadd -g 1000 kbkn && useradd -m -s /bin/bash -u 1000 -g kbkn kbkn; fi && echo "kbkn:kobaken" | chpasswd && mkdir -p /home/kbkn/work /home/kbkn/.local/bin /home/kbkn/.config/pip && printf "[global]\nbreak-system-packages = true\n" > /home/kbkn/.config/pip/pip.conf && PYTHON_CMD=$(which python3 || echo "python3") && $PYTHON_CMD -m pip install --no-cache-dir --break-system-packages jupyterlab jupyterhub scikit-learn matplotlib pandas transformers ipywidgets ultralytics && (if [ -f /opt/conda/bin/conda ]; then /opt/conda/bin/conda clean -a -y; fi) && ($PYTHON_CMD -m pip cache purge || true) && rm -rf /root/.cache/pip /home/kbkn/.cache/pip /tmp/* && chown -R kbkn:kbkn /home/kbkn && if [ -d "/opt/conda" ]; then chown -R kbkn:kbkn /opt/conda; fi # buildkit |
| ENV PATH=/home/kbkn/.local/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV PIP_BREAK_SYSTEM_PACKAGES=1 |
| WORKDIR /home/kbkn/work |
| USER kbkn |
| ENTRYPOINT [] |
