Installation¶
GraphNeT is available for Python 3.8 to Python 3.11.
Note
We recommend installing GraphNeT in a separate environment, e.g. using a Python virtual environment or Anaconda (see details on installation here).
Quick Start¶
PyTorch
Your OS
CUDA
Run:
When installation is completed, you should be able to run the examples.
Installation in CVMFS (IceCube)¶
You may want GraphNeT to be able to interface with IceTray, e.g., when converting I3 files to a deep learning friendly file format, or when deploying models as part of an IceTray chain. In these cases, you need to install GraphNeT in a Python runtime that has IceTray installed.
To achieve this, we recommend installing GraphNeT into a CVMFS with IceTray installed, like so:
# Download GraphNeT
git clone https://github.com/graphnet-team/graphnet.git
cd graphnet
# Open your favorite CVMFS distribution
eval `/cvmfs/icecube.opensciencegrid.org/py3-v4.2.1/setup.sh`
/cvmfs/icecube.opensciencegrid.org/py3-v4.2.1/RHEL_7_x86_64/metaprojects/icetray/v1.5.1/env-shell.sh
# Update central utils
pip install --upgrade pip>=20
pip install wheel setuptools==59.5.0
# Install graphnet into the CVMFS as a user
pip install --user -r requirements/torch_cpu.txt -e .[torch,develop]
Once installed, GraphNeT is available whenever you open the CVMFS locally.