from nvidia/cuda:12.4.1-devel-ubuntu22.04
from python:3.12

env DEBIAN_FRONTEND=noninteractive

run apt-get update && apt-get install -y \
    git \
    llvm \
    ffmpeg \
    libsm6 \
    libxext6 \
    ninja-build

# env CC /usr/bin/clang
# env CXX /usr/bin/clang++
# 
# # install llvm10 as required by llvm-lite
# run git clone https://github.com/llvm/llvm-project.git -b llvmorg-10.0.1
# workdir /llvm-project
# # this adds a commit from 12.0.0 that fixes build on newer compilers
# run git cherry-pick -n b498303066a63a203d24f739b2d2e0e56dca70d1
# run cmake -S llvm -B build -G Ninja -DCMAKE_BUILD_TYPE=Release
# run ninja -C build install  # -j8

run curl -sSL https://install.python-poetry.org | python3 -

env PATH "/root/.local/bin:$PATH"

copy . /skynet

workdir /skynet

env POETRY_VIRTUALENVS_PATH /skynet/.venv

run poetry install --with=cuda -v

workdir /root/target

env PYTORCH_CUDA_ALLOC_CONF max_split_size_mb:128
env NVIDIA_VISIBLE_DEVICES=all

copy docker/entrypoint.sh /entrypoint.sh
entrypoint ["/entrypoint.sh"]

cmd ["skynet", "--help"]