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