From 6fc6c3a623442971b3117dcf174227db24c7a38c Mon Sep 17 00:00:00 2001 From: hy3ol <0929khy@gmail.com> Date: Mon, 23 Mar 2026 18:22:25 +0900 Subject: [PATCH] Add paper reproduction requirements --- requirements.txt | 9 +++++++++ 1 file changed, 9 insertions(+) create mode 100644 requirements.txt diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..5d770af --- /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,<2.0 +pandas>=1.5,<3.0 +scikit-learn>=1.3,<2.0 +einops>=0.7,<1.0 +tqdm>=4.66,<5.0