文章预览
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
………………………………