Lumi-Transformer / requirements.txt
l-o0o-k's picture
Upload requirements.txt
aff2585
python==3.7.16
torch==1.9.0+cu111
torchvision==0.10.0+cu111
cudatoolkit==10.2.89
cuda==12.1.1
timm==0.4.12
opencv-python==4.4.0.46
termcolor==1.1.0
yacs==0.1.8
pyyaml==6.0
scipy==1.7.3