--extra-index-url https://download.pytorch.org/whl/cu124

torch==2.6.0+cu124
scikit-learn==1.6.1
pandas>=2.1.4
opt_einsum==3.4.0