Google Colab GPU not working

Due to the most recent release of pytorch installing the proper torchwheels based on your CUDA version isn’t quite automated yet for colab. As a result I would recommend using the following to install fastai and pytorch:

!pip install light-the-torch >> /.tmp
!ltt install torch torchvision >> /.tmp
!pip install fastai --upgrade >> /.tmp

(Run that in a cell)

6 Likes