Install pytorch

Follow the guide on https://pytorch.org/get-started/locally/

Create virtual environment

Via conda:

conda env create -f environment.yml

Via pip:

pip install -r requirements.txt