colab mount drive from google.colab import drive drive.mount('/content/gdrive') install pytorch 1.2 !pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html install pytorch apex