ai models graphcast

ai-model-graphcast

ai-models-graphcast adalah plugin ai-models untuk menjalankan GraphCast Google Deepmind.

GraphCast: Mempelajari prakiraan cuaca global jarak menengah yang terampil, pracetak arXiv: 2212.12794, 2022. https://arxiv.org/abs/2212.12794

GraphCast dibuat oleh 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, Jacklynn Stott, Alexander Pritzel, Shakir Mohamed dan Peter Battaglia.

Bobot model tersedia untuk digunakan berdasarkan ketentuan Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0). Anda dapat memperoleh salinan Lisensi di: https://creativecommons.org/licenses/by-nc-sa/4.0/.

Instalasi

Untuk menginstal paket, jalankan:

 pip instal ai-model-graphcast

Ini akan menginstal paket dan sebagian besar dependensinya.

Kemudian untuk menginstal dependensi graphcast (dan Jax pada GPU):

Graphcast dan Jax

Graphcast bergantung pada Jax, yang memerlukan instruksi instalasi khusus untuk perangkat keras spesifik Anda.

Silakan lihat panduan instalasi untuk mengikuti instruksi yang benar.

Kami telah menyiapkan dua requirements.txt yang dapat Anda gunakan. Versi CPU dan GPU:

Untuk penggunaan GPU pilihan:

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

Untuk penggunaan CPU yang lebih lambat:

pip install -r requirements.txt