singularity.def

BootStrap: docker
From: scilus/scilus:1.3.0

%files
    /home/fdumais/Desktop/Acuity/tractolearn /tractolearn
    /usr/share/fonts/truetype/freefont/FreeSans.ttf /usr/share/fonts/truetype/freefont/FreeSans.ttf

%post
    apt install -y jq

    pip install --upgrade pip

    pip install --upgrade vtk==9.1.*
    pip install --upgrade dmri-amico==1.2.*
    pip install --upgrade pyparsing==2.2.*
    pip install --upgrade cycler==0.10.*
    pip install --upgrade kiwisolver==1.0.*
    pip install --upgrade python-dateutil==2.7.*
    pip install --upgrade pytz==2018.4.*
    pip install --upgrade setuptools==46.1.*
    pip install --upgrade six==1.15.*

    cd tractolearn
    pip install -r requirements.txt
    pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html
    pip install -e .

    chmod +x scripts/filter_streamline.py