Bootstrap: docker
From: python:3.10
%post 
    apt-get -y update
    python -m pip install --upgrade pip
    pip install matplotlib numpy pandas scikit-learn sympy tqdm scipy einops
    pip install time-interpret captum
    pip install torch==2.2 --index-url https://download.pytorch.org/whl/cu118
    pip install pytorch-lightning==1.8.6 \\
        ipykernel \
        ipywidgets \
        jupyter_client \
        notebook \
        numpy \
        setuptools \
        reformer-pytorch

%labels
    Author TimeSeries
    Version v1.0

%help
    A container for Time Series Forecasting.