torch==2.1.2+cu118 torchvision==0.16.2+cu118 xformers==0.0.23.post1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 # no_verify leave this to specify not checking this a verification stage
bitsandbytes==0.43.0
tensorboard==2.14.1 tensorflow==2.14.0 wheel
tensorrt
onnxruntime-gpu==1.17.1
-r requirements.txt