fumet.def

# Copyright Genki Kinoshita, 2024. All rights reserved.
#
# This file is part of a software package that is based on the Monodepth2
# software, originally developed by Niantic. and distributed under the
# Monodepth2 license. The terms of the Monodepth2 license apply to any usage
# of this file in conjunction with the original Monodepth2 software.

Bootstrap: docker
From: nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu18.04

%post
    export DEBIAN_FRONTEND=noninteractive
    export DEBCONF_NONINTERACTIVE_SEEN=true
    export PYTHONNOUSERSITE=1

    rm -rf /var/lib/apt/lists/*
    apt-get -y update
    apt-get -y install vim git wget
    apt-get -y install make build-essential libssl-dev zlib1g-dev \
        libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
        libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev

    # install pyenv
    export PYENV_ROOT=/opt/pyenv
    export PATH=$PYENV_ROOT/bin:$PYENV_ROOT/shims:${PATH}
    export PYTHON_VERSION=3.10.4
    git clone https://github.com/pyenv/pyenv.git /opt/pyenv
    pyenv install ${PYTHON_VERSION}
    pyenv global ${PYTHON_VERSION}

    pip install --upgrade --no-cache-dir pip

    pip --no-cache-dir install torch==2.0.1 torchvision==0.15.2
    pip --no-cache-dir install pillow Pillow==8.4.0 protobuf==3.19.1 timm==0.4.12
    pip --no-cache-dir install tensorboard
    pip --no-cache-dir install opencv-python
    pip --no-cache-dir install matplotlib
    pip --no-cache-dir install scikit-image
    pip --no-cache-dir install tqdm

    pip install 'git+https://github.com/saadnaeem-dev/pytorch-linear-warmup-cosine-annealing-warm-restarts-weight-decay'

    pip --no-cache-dir install -U numpy==1.26.4

%environment
    export PYTHONNOUSERSITE=1
    export PYENV_ROOT=/opt/pyenv
    export PATH=$PYENV_ROOT/bin:$PYENV_ROOT/shims:${PATH}
    export PYTHON_VERSION=3.10.4