Heads up for Colab users:
To use PyTorch 1.6 in colab, you need to do the following (and when installing fastai):
!wget 'torch-1.6.0+cu101-cp36-cp36m-linux_x86_64.whl'
!pip install 'torch-1.6.0+cu101-cp36-cp36m-linux_x86_64.whl'
(then of course pip install fastai2
, etc)
If you’re running CUDA 10.2 then you will need to replace the whl
link (and filename) with:
https://download.pytorch.org/whl/cu102/torch-1.6.0-cp36-cp36m-linux_x86_64.whl
and torch-1.6.0-cp36-cp36m-linux_x86_64.whl