Tsubatoの発信記録

主に機械学習やデータサイエンス関連で学んだことを書いています。

google colabのTPUをPyTorchで使う

問題

  • google colabのTPUをPytorchで使いたい。
  • ググって出てくるセットアップをするとエラーが出る。
# !pip install cloud-tpu-client==0.10 https://storage.googleapis.com/tpu-pytorch/wheels/torch_xla-1.9-cp37-cp37m-linux_x86_64.whl
ERROR: torch_xla-1.9-cp37-cp37m-linux_x86_64.whl is not a supported wheel on this platform.

解決方法

  • 闇雲にググらないで公式のgithubを見る。

github.com

  • colabでの動かし方は↓のnotebookにあります。

colab.research.google.com

学び

  • 大体のことは適当にググって上の方に出てくる方法で解決するけど、インストール関連など公式ページを見たほうが良いトピックもあるので気をつけよう。