TLDR; here is the link to the notebook — https://drive.google.com/open?id=18R3Rz639Fa4ByFwzLrY5LmD8t3Ee5IUR
In Lesson 9 of Udacity’s Deep Learning with Python, we explore the latest features of PyTorch. So, to get the appropriate version working on Google Colab, I made this handy notebook, which installs correct PyTorch version based on the GPU architecture of Google Colab.
Here are the brief steps:
- Download the modified Notebook from this link
- Visit Google Colab
- You will be prompted with a modal, select Upload
- If modal doesn’t appear and instead if it opens a new Notebook, then from menu File > Upload Notebook
- Upload the modified notebook
- That’s it!
If you are curious to know what I did, then I wrote this following script which downloads PyTorch v1 stable:
!pip install -q torch==1.0.0 torchvision
Hope this helps! If you have got any issues with running this notebook, then feel free to message me on Slack for any help. My slack username is avinashss
All the best for the last lesson!