base.Dockerfile

# Build arguments
ARG CUDA_VER=13.0.2
ARG UBUNTU_VER=24.04
# Download the base image
FROM nvidia/cuda:${CUDA_VER}-cudnn-runtime-ubuntu${UBUNTU_VER}
# you can check for all available images at https://hub.docker.com/r/nvidia/cuda/tags
# Install as root
USER root
# Shell
SHELL ["/bin/bash", "--login", "-o", "pipefail", "-c"]
ARG ZELLIJ_VERSION=v0.40.1

# Install dependencies
ARG DEBIAN_FRONTEND="noninteractive"
RUN apt-get update && apt-get install -y --no-install-recommends \
    bash \
    bash-completion \
    ca-certificates \
    curl \
    git \
    htop \
    nano \
    nvidia-modprobe \
    nvtop \
    openssh-client \
    python3 python3-dev python-is-python3 \
    sudo \
    tmux \
    unzip \
    vim \
    wget \
    zip && \
    apt-get autoremove -y && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*

# Download and install zellij
RUN curl -L -o zellij.tar.gz "https://github.com/zellij-org/zellij/releases/download/${ZELLIJ_VERSION}/zellij-x86_64-unknown-linux-musl.tar.gz" && \
    tar -xzf zellij.tar.gz -C /usr/local/bin && \
    rm zellij.tar.gz && \
    zellij --version

# Change to your user
USER ubuntu
# Chnage Workdir
WORKDIR /home/ubuntu
# Download and install uv
RUN curl -LsSf https://astral.sh/uv/install.sh | sh
ENV PATH="/home/ubuntu/.local/bin:/home/ubuntu/.cargo/bin:${PATH}"
USER root
# Install packages inside the new environment
RUN uv pip install --no-cache --system --break-system-packages --upgrade pip setuptools wheel && \
    uv pip install --no-cache --system --break-system-packages \
    ipywidgets \
    jupyterlab \
    matplotlib \
    nltk \
    notebook \
    numpy \
    pandas \
    Pillow \
    plotly \
    PyYAML \
    scipy \
    scikit-image \
    scikit-learn \
    sympy \
    seaborn \
    tqdm && \
    find /usr/local/lib/python3.* -name '__pycache__' -exec rm -rf {} + && \
    # Set path of python packages
    echo "# Set path of python packages" >>/home/ubuntu/.bashrc && \
    echo "export PATH=\$HOME/.local/bin:\$PATH" >>/home/ubuntu/.bashrc
# Change to your user
USER ubuntu