Heads up for Colab users:
To use PyTorch 1.6 in colab, you need to do the following (and when installing fastai):
!pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
(then of course pip install fastai2, etc)
If you’re running CUDA 10.2 then you just need to do !pip install torch torchvision --upgrade