NeuralGCM气象模型上手教程

文章预览

NeuralGCM模型进行气象预测教程 简介 NeuralGCM （General circulation models）是Google开发的一种新型的天气和气候模型，它结合了 传统的物理建模与机器学习技术 ，相关论文于2024年发表在nature上。它 既可以做短期的天气预报，也可以做长期的气候预测 。 安装以及初始化： 我的环境是 在windows11上的WSL 2.0 Ubuntu虚拟机， ，显卡为4070，cuda版本为12.6， conda里的 python版本为3.10，安装步骤如下： conda create --name neural_gcm_env python=3.10 pip install neuralgcm pip install gcsfs pip install  "jax[cuda126]"  -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html 安装完毕后进行环境测试： （1）测试gcsfs import  gcsfs import  jax import  numpy  as  np import  pickle import  xarray from  dinosaur  import  horizontal_interpolation from  dinosaur  import  spherical_harmonic from  dinosaur  import  xarray_utils import  neural ………………………………