Colab上のPyMC5をGPUで動かす - Qiita

はじめに 『Pythonでスラスラわかる ベイズ推論「超」入門』著者の赤石です。 本書の実習のうち、後ろの方の実習は、sample関数呼び出しに相当の時間がかかります。 その対策として、6.3節のコラムに変分推論法の利用について書いているわけですが、他に手はないのだろうかと思っていました。 条件によってはGPU利用が有効であることがわかったので、その方法を説明することにします。 取り上げる例題は、6.3節の『IRT(Item Response Theory)によるテスト評価』です。 手順 Colab上でGPUを利用してsample関数を呼び出すための手順は、以下になります。 ランタイムのタイプをGPUに変更 ライブラリの更新・追加導入 sample関数呼び出し時のオプション追加 では、早速、実際に試してみましょう。 ランタイム環境の変更 Colabのランタイム環境をGPUに変える手順は通常