ai models graphcast

ai-models-graphcast

ai-models-graphcast ist ein Ai-Models-Plugin zum Ausführen von GraphCast von Google Deepmind.

GraphCast: Erlernen geschickter mittelfristiger globaler Wettervorhersagen, arXiv-Vorabdruck: 2212.12794, 2022. https://arxiv.org/abs/2212.12794

GraphCast wurde von 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 erstellt. Jacklynn Stott, Alexander Pritzel, Shakir Mohamed und Peter Battaglia.

Die Modellgewichtungen werden gemäß den Bedingungen von Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0) zur Nutzung zur Verfügung gestellt. Eine Kopie der Lizenz erhalten Sie unter: https://creativecommons.org/licenses/by-nc-sa/4.0/.

Installation

Um das Paket zu installieren, führen Sie Folgendes aus:

 pip install ai-models-graphcast

Dadurch werden das Paket und die meisten seiner Abhängigkeiten installiert.

Dann installieren Sie Graphcast-Abhängigkeiten (und Jax auf der GPU):

Graphcast und Jax

Graphcast ist auf Jax angewiesen, das spezielle Installationsanweisungen für Ihre spezifische Hardware benötigt.

Bitte beachten Sie die Installationsanleitung, um die korrekten Anweisungen zu befolgen.

Wir haben zwei requirements.txt vorbereitet, die Sie verwenden können. Eine CPU- und eine GPU-Version:

Für die bevorzugte GPU-Nutzung:

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

Für die langsamere CPU-Auslastung:

pip install -r requirements.txt