diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..01c421e --- /dev/null +++ b/requirements.txt @@ -0,0 +1,9 @@ +# Recommended Python: 3.10.x +# RTX 5090 / Blackwell requires a CUDA 12.8+ PyTorch build. +--extra-index-url https://download.pytorch.org/whl/cu128 +torch==2.9.0+cu128 +numpy>=1.26,<3.0 +pandas>=1.5,<3.0 +scikit-learn>=1.3,<2.0 +einops>=0.7,<1.0 +tqdm>=4.66,<5.0