ai models graphcast

ai-模型-graphcast

ai-models-graphcast是一個 ai-models 插件，用於運行 Google Deepmind 的 GraphCast。

GraphCast：學習熟練的中期全球天氣預報，arXiv 預印本：2212.12794，2022。

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 創建，傑克琳·斯托特、亞歷山大·普里策爾、沙基爾·穆罕默德和彼得·巴塔利亞。

模型權重可依據 Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0) 的條款使用。您可以在以下位置取得授權副本：https://creativecommons.org/licenses/by-nc-sa/4.0/。

安裝

若要安裝軟體包，請執行：

 pip 安裝 ai-models-graphcast

這將安裝該套件及其大部分依賴項。

然後安裝 graphcast 依賴項（以及 GPU 上的 Jax）：

Graphcast 和 Jax

Graphcast 依賴 Jax，它需要針對您的特定硬體的特殊安裝說明。

請參閱安裝指南以遵循正確的說明。

我們準備了兩個requirements.txt供您使用。 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