# torch==2.2.0
# torchvision==0.17.0+cu118
# transformers==4.37.2