ai models graphcast

ai-モデル-グラフキャスト

ai-models-graphcast Google Deepmind の GraphCast を実行するための ai-models プラグインです。

GraphCast: 巧みな中距離地球規模天気予報の学習、arXiv プレプリント: 2212.12794、2022。 https://arxiv.org/abs/2212.12794

GraphCast は、Remi Lam、Alvaro Sanchez-Gonzalez、Matthew Willson、Peter Wirnsberger、Meire Fortunato、Ferran Alet、Suman Ravuri、Timo Ewalds、Zach Eaton-Rosen、Weihua Hu、Alexander Merose、Stephan Hoyer、George Holland、Oriol Vinyals によって作成されました。ジャックリン・ストット、アレクサンダー・プリッツェル、シャキールモハメドとピーター・バタグリア。

モデルの重みは、クリエイティブ コモンズ表示 - 非営利 - 継承 4.0 インターナショナル (CC BY-NC-SA 4.0) の条件に基づいて使用できるようになります。ライセンスのコピーは、https://creativecommons.org/licenses/by-nc-sa/4.0/ で入手できます。

インストール

パッケージをインストールするには、次を実行します。

 pip インストール ai-models-graphcast

これにより、パッケージとその依存関係のほとんどがインストールされます。

次に、グラフキャストの依存関係 (および GPU 上の Jax) をインストールします。

グラフキャストとジャックス

Graphcast は Jax に依存しているため、特定のハードウェアに特別なインストール手順が必要です。

正しい手順に従ってインストール ガイドを参照してください。

ご利用いただけるrequirements.txt 2つ用意しました。 CPU と GPU のバージョン:

推奨される GPU 使用率については、次のとおりです。

pip install -r requirements-gpu.txt -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

CPU 使用率が遅い場合:

pip install -r requirements.txt