Installation
Requirements
- Python 3.9+
- PyTorch 2.0+
From PyPI
GPU support
PyTorch will default to CPU. For GPU training, install torch with CUDA first:
# CUDA 11.8
pip install torch --index-url https://download.pytorch.org/whl/cu118
pip install gp-tempest
# CUDA 12.1
pip install torch --index-url https://download.pytorch.org/whl/cu121
pip install gp-tempest