Singularity.v3.8-torch1.5.0-dj0.12.4

BootStrap: library
From: ubuntu:20.04

%labels
MAINTAINER Christoph Blessing <chris24.blessing@gmail.com>

%post
    apt update && apt install -y software-properties-common
    add-apt-repository universe

    apt update && apt install -y \
        build-essential \
        git \
        wget \
        vim \
        curl \
        zip \
        zlib1g-dev \
        unzip \
        pkg-config \
        libblas-dev \
        liblapack-dev \
        python3-tk \
        python3-wheel \
        graphviz \
        libhdf5-dev \
        python3.8 \
        python3.8-dev \
        python3.8-distutils \
        swig
    apt-get clean

    ln -s /usr/bin/python3.8 /usr/local/bin/python
    ln -s /usr/bin/python3.8 /usr/local/bin/python3

    curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
    python3.8 get-pip.py
    rm get-pip.py

    python3.8 -m pip --no-cache-dir install \
        blackcellmagic\
        pytest \
        pytest-cov \
        numpy \
        matplotlib \
        scipy \
        pandas \
        jupyter \
        scikit-learn \
        scikit-image \
        seaborn \
        graphviz \
        gpustat \
        h5py \
        gitpython \
        Pillow==6.2.0 \
        torch==1.5.0 \
        torchvision==0.6.0 \
        jupyterlab \
        datajoint==0.12.4

%environment
    export SHELL=/bin/sh

%runscript
    jupyter lab