1.新建conda环境

conda create -n PyG python=3.8

2.安装pytorch环境

官网:PyTorch

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

3.安装PyG,torch_geometric

官网:安装 — pytorch_geometric 文档 (pytorch-geometric.readthedocs.io)

pip install torch_geometric
或
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.4.0+cu121.html

4.安装xarray环境

官网:Installation (xarray.dev)

conda install -c conda-forge xarray dask netCDF4 bottleneck

5.安装Cartopy

详见:Installing Cartopy — cartopy 0.19.0rc2.dev8+gd251b2f documentation (scitools.org.uk)

conda install -c conda-forge cartopy

6.安装jupyter

官网:Project Jupyter | Installing Jupyter

pip install jupyter

7.tensorboard

pip install tensorboard
08-25 08:53