This is just a version issue...I did
!pip install torch==1.4 torchvision==0.14.1
and it seems to work fine!