/hgcal_ldrd

Code repository for HGCal LDRD

Primary LanguageJupyter NotebookBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

hgcal_ldrd

Code repository for HGCal LDRD

You will need to:

conda create --name hgcal-env python=3.6
conda activate hgcal-env
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
conda install pandas matplotlib jupyter nbconvert==5.4.1
conda install -c conda-forge tqdm
pip install uproot scipy sklearn --user
pip install torch-scatter torch-sparse
pip install networkx

(or replace pip with the corresponding conda installation for safer compatibility)

and install pytorch geometric according to the instructions here: https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html