NVIDIA / modulus-launch

Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems
Apache License 2.0
56 stars 27 forks source link

🐛[BUG]: ModuleNotFoundError: No module named 'warp' #126

Closed sangnguyens closed 1 year ago

sangnguyens commented 1 year ago

Version

[0.4.0] - 2023-11-xx

On which installation method(s) does this occur?

Pip, Source

Describe the issue

running darcy_fno model I got the error below: Traceback (most recent call last): File "/media/adminisatrator/SSD1/nvidia/self-peace/darcy_fno_modulus_launch/train_darcy_fno.py", line 12, in <module> from modulus.datapipes.benchmarks.darcy import Darcy2D File "/home/adminisatrator/.local/lib/python3.10/site-packages/modulus/datapipes/benchmarks/darcy.py", line 18, in <module> import warp as wp ModuleNotFoundError: No module named 'warp'

Can anyone help to fix it?

Minimum reproducible example

No response

Relevant log output

No response

Environment details

No response

sangnguyens commented 1 year ago

I missed package warp-lang as mentioned in readme.