I am using pytorch 1.18.5 (pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu101/torch_nightly.html) and running my code using google colab.<