如何在colab用pytorch_lightning白嫖TPU
如何在colab用pytorch_lightning白嫖TPU
首先
我们得熟悉pytorch_lightning
以及colab
的操作。在使用TPU之前,先挂载好我们的云端磁盘以及配置好pytorch需要的TPU环境。
1 | from google.colab import drive |
安装好pytorch_lightning
以及配置好环境之后,我们就可以开始在pytorch_lightning
框架下跟写GPU代码一样写TPU代码。下面请看VCR,一个小栗子。