运行earthformer环境
python -m pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 -f https://download.pytorch.org/whl/torch_stable.html
python -m pip install pytorch_lightning==1.6.4
python -m pip install xarray netcdf4 opencv-python
我的numpy的版本是1.20.0,更新命令
pip install --upgrade numpy==1.20.0
其他的包版本更新安装
按照提示,更新一些版本,有的版本太高了需要降
pip install --upgrade XXX==xxx
安装apex
直接pip install apex不行会出错。所以以下操作:
将git下来的文件,将apex放到earthformer1里
运行git clone https://www.github.com/nvidia/apex
cd apex
python -m pip install -U -e . --no-build-isolation
参考:https://blog.csdn.net/qq_37457202/article/details/129038437