ai models graphcast

ai-models-graphcast

ai-models-graphcast เป็นปลั๊กอิน ai-models เพื่อเรียกใช้ GraphCast ของ Google Deepmind

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

สิ่งนี้จะติดตั้งแพ็คเกจและการขึ้นต่อกันส่วนใหญ่

จากนั้นเพื่อติดตั้งการพึ่งพากราฟคาสต์ (และ Jax บน GPU):

กราฟแคสต์และแจ็กซ์

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