earth2studio icon indicating copy to clipboard operation
earth2studio copied to clipboard

🐛[BUG]: SFNO/Makani missing some needed deps and fails on vanilla Pytorch container

Open NickGeneva opened this issue 1 year ago • 0 comments

Version

main

On which installation method(s) does this occur?

Pip, Source

Describe the issue

Present the SFNO model requires makani install.

This is inside nvcr.io/nvidia/pytorch:24.01-py3

The current deps of makani and also whats in e2studio are missing some items. Not sure what triggered this. I think something got updated in the Modulus version 0.6.0 that removed some deps that Makani assumed would be installed.

Packages needed to get added are:

pip install ruamel.yaml
pip install torch-harmonics
pip install tensorly
pip install tensorly-torch

```

NickGeneva avatar May 16 '24 05:05 NickGeneva