Just wanted to know if Fast.ai supports TPUs if I run Jupyter notebooks in Google Colab environment?
2 Likes
Not to my knowledge. TPUs work with Tensorflow and not very well with PyTorch. Supposedly, integration with PyTorch is in the works. Then fastai should be able to use them, too.
1 Like
Similar discussion on PyTorch forum. https://discuss.pytorch.org/t/pytorch-tpu-support/25504
2 Likes
Is fastAI2 compatible with TPU on Google Colab?
No it is not at the moment
1 Like