torch==1.8.0
torchvision==0.9.0
opencv-python==4.5.2.54
timm==0.4.5
torch==1.8.1
torchvision==0.9.1
opencv-python==4.5.2.54
timm==0.4.5

Upgrade Pytorch version to circumvent CUDNN issues
Upgrade Pytorch version to circumvent CUDNN issues