Dockerfile

# Use CUDA 12.4
FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
SHELL ["/bin/bash", "-c"]
ENV SHELL=/bin/bash
ENV DEBIAN_FRONTEND=noninteractive

# Export paths for CUDA and cuDNN
ENV CUDA_HOME=/usr/local/cuda
ENV PATH="/usr/local/cuda/bin:$PATH"
ENV CPATH="/usr/local/cuda/include:/usr/include:$CPATH"
ENV LIBRARY_PATH="/usr/local/cuda/lib64:$LIBRARY_PATH"
ENV LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH"

RUN echo 'command -v nvcc &>/dev/null && nvcc -V || uname -srm' >> /root/.bashrc && \
    echo 'eval "$(starship init bash)"' >> /root/.bashrc

# Install basic packages
RUN apt-get update --fix-missing && \
    apt-get install bzip2 ca-certificates curl wget git vim nano tree -y && \
    apt-get install texlive-xetex texlive-fonts-recommended texlive-plain-generic -y && \
    apt-get clean && rm -rf /var/lib/apt/lists/*

# Install Conda Environment
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/install.sh && \
    bash ~/install.sh -b -p /root/conda && rm ~/install.sh && \
    eval "$('/root/conda/bin/conda' 'shell.bash' 'hook')" && conda init

# Create Development Environment
ENV PATH="/root/conda/bin:$PATH"
RUN conda create -n lab python=3.12 anaconda -y && \
    conda clean -a && pip cache purge

# Setup JupyterLab plugins
RUN eval "$('/root/conda/bin/conda' 'shell.bash' 'hook')" && conda activate lab && \
    conda install -c conda-forge starship nodejs pandoc texlive-core texlab chktex && \
    pip install 'jupyterlab>=4.1.0,<5.0.0a0' jupyterlab-lsp 'python-lsp-server[all]' \
        jupyterlab-code-formatter black isort jupyterlab-spellchecker jupyterlab-latex \
        jupyter-resource-usage jupyterlab_execute_time lckr_jupyterlab_variableinspector \
        jupyterlab-git jupyterlab_wakatime 'ipywidgets>=8.0' && \
    npm set prefix /root && npm install -g --save-dev remark-language-server \
        remark-preset-lint-consistent remark-preset-lint-recommended && \
    jupyter labextension disable "@jupyterlab/apputils-extension:announcements" && \
    conda clean -a && pip cache purge && npm cache clean --force

COPY JupyterLabConfig/jupyter_lab_config.py /root/.jupyter/jupyter_lab_config.py
COPY JupyterLabConfig/extensions/ /root/.jupyter/lab/user-settings/\@jupyterlab/
COPY JupyterLabConfig/jupyterlab-lsp/ /root/.jupyter/lab/user-settings/\@jupyter-lsp/jupyterlab-lsp/
COPY JupyterLabConfig/jupyterlab-lsp/unified_language_server.py /root/conda/envs/lab/lib/python3.12/site-packages/jupyter_lsp/specs/unified_language_server.py
COPY JupyterLabConfig/jupyterlab-lsp/remarkrc.yml /root/.remarkrc.yml
COPY JupyterLabConfig/notebooks/ /root/projects/demo_notebooks/
COPY JupyterLabConfig/starship.toml /root/.config/starship.toml

# Install PyTorch and AI libs
RUN eval "$('/root/conda/bin/conda' 'shell.bash' 'hook')" && conda activate lab && \
    pip install torch torchvision torchaudio transformers datasets accelerate peft && \
    conda clean -a && pip cache purge

# Run JupyterLab on start
ENV DEBIAN_FRONTEND=dialog
RUN echo 'conda activate lab' >> /root/.bashrc && \
    echo 'jupyter lab' > /root/entrypoint.sh

WORKDIR /root/projects
ENTRYPOINT ["/bin/bash", "-i", "/root/entrypoint.sh"]
EXPOSE 80

HEALTHCHECK --interval=30s --timeout=10s --retries=3 \
CMD curl -f -s http://localhost:80/lab || exit 1