GPU/TPU スイッチの導入

[kabupen]

• GPU は週30h
• TPU は週20h なので、これらを合わせて使用するために、コードをTPU対応にする。

[kabupen]

https://qiita.com/K4me/items/a8137a55954ac5425363

[kabupen]

Kaggle でうまく行かない？

• Acceralator を「TPU」に設定

  !pip uninstall -y torch torchvision fastai easyocr allennlp torchtext torchaudio pytorch-lightning kornia fairscale
  !curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py
  !python pytorch-xla-env-setup.py --apt-packages libomp5 libopenblas-dev
  !rm *.whl pytorch-xla-env-setup.py
  !pip install torch_optimizer

  を実行した後、

  import torch

  がまず not found になる

[kabupen]

!curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py
!python pytorch-xla-env-setup.py --version nightly --apt-packages libomp5 libopenblas-dev

に変更してみた