创建环境
根据github中的代码GitHub - Aalto-QuML/ClimODE: ClimODE: Climate and Weather Forecasting With Physics-informed Neural ODEs
创建环境climode,这里用的是
conda create --name climode python=3.8
安装如下这些库
- torchdiffeq : GitHub - rtqichen/torchdiffeq: Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation..
- pytorch >= 1.12.0
- torch-scatter
- torch-sparse
- torch-cluster
- torch-spline-conv
- torchcubicspline: GitHub - patrick-kidger/torchcubicspline: Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, and backpropagation.
- properscoring (for CRPS scores) : properscoring · PyPI