ai models graphcast

ai-模型-graphcast

ai-models-graphcast是一个 ai-models 插件，用于运行 Google Deepmind 的 GraphCast。

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 创建，杰克琳·斯托特、亚历山大·普里策尔、沙基尔·穆罕默德和彼得·巴塔利亚。

模型权重可根据 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