0 快速环境
# 建议 CUDA 12 + PyTorch 2.3
conda create -n tc-track python=3.11 pytorch cudatoolkit=12.1 -c pytorch -y
conda activate tc-track
pip install --upgrade xarray netCDF4 einops geopy shapely cartopy tqdm wandb \
torchmetrics h5py requests cdsapi
# ↓ GraphCast & Pangu 推理代码(任选其一或两者并行)
pip install ai-models-graphcast ai-models-panguweather # ECMWF 封装器 :contentReference[oaicite:0]{index=0}
1 数据管线
数据源 |
变量 |
用途 |
获取示例 |
IBTrACS v04 |
位置信息(lat, lon, pmin, vmax…) |
监督标签 |
wget https://www.ncei.noaa.gov/.../ibtracs.ALL.v04r00.csv.gz |