requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu121
--find-links https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.4.0_cu121.html