ecmwf-aifs-space / requirements.txt
saburq's picture
feat: Add CPU support and improve device handling
30ffd75
raw
history blame contribute delete
304 Bytes
# For CPU-only installation, use:
torch
# For CUDA installation, use:
# --extra-index-url https://download.pytorch.org/whl/cu118
# torch==2.0.1+cu118
flash-attn
anemoi-inference[huggingface]==0.4.9
anemoi-models==0.3.1
earthkit-regrid==0.4.0
ecmwf-opendata
gradio
numpy
matplotlib
cartopy
xarray
netCDF4