# Pre-configured requirements file to automatically resolve PyTorch with GPU/CUDA 12.1 support on Windows/Linux.
# For automated GPU/CUDA/MPS detection, prefer: python install_gpu.py
# Run: pip install -r requirements-gpu.txt

--extra-index-url https://download.pytorch.org/whl/cu121
torch>=2.0
pyro-ppl>=1.9
-e .[dev,viz,nf,hmc]
