I used following for installation:
!pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html


Now next time, if I get by chance A100, I will check it out. 
Thanks again, for helping me out