torch==2.1.2+cu118 --index-url https://download.pytorch.org/whl/cu118
torchvision==0.16.2+cu118 --index-url https://download.pytorch.org/whl/cu118
xformers==0.0.23.post1+cu118 --index-url https://download.pytorch.org/whl/cu118