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, Jacklynn Stott, Alexander Pritzel, Shakir Mohamed 및 Peter Battaglia.

모델 가중치는 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-모델-그래프캐스트

그러면 패키지와 해당 종속성 대부분이 설치됩니다.

그런 다음 그래프캐스트 종속 항목(및 GPU에 Jax)을 설치하려면 다음을 수행하세요.

그래프캐스트와 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