OS: Ubuntu 22.04
CUDA version: 12.5
I had pulled the image nvcr.io/nvidia/nemo:24.09 from NVIDIA, then started a container and ran the DreamFusion example code from NVIDIA: NeMo/examples/multimodal/x_to_nerf at main · NVIDIA/NeMo.
However, I got this error:
Error executing job with overrides: []
Traceback (most recent call last):
File "/workspace/DreamFusion/main.py", line 70, in <module>
main()
File "/opt/NeMo/nemo/core/config/hydra_runner.py", line 129, in wrapper
_run_hydra(
File "/usr/local/lib/python3.10/dist-packages/hydra/_internal/utils.py", line 394, in _run_hydra
_run_app(
File "/usr/local/lib/python3.10/dist-packages/hydra/_internal/utils.py", line 457, in _run_app
run_and_report(
File "/usr/local/lib/python3.10/dist-packages/hydra/_internal/utils.py", line 223, in run_and_report
raise ex
File "/usr/local/lib/python3.10/dist-packages/hydra/_internal/utils.py", line 220, in run_and_report
return func()
File "/usr/local/lib/python3.10/dist-packages/hydra/_internal/utils.py", line 458, in <lambda>
lambda: hydra.run(
File "/usr/local/lib/python3.10/dist-packages/hydra/_internal/hydra.py", line 132, in run
_ = ret.return_value
File "/usr/local/lib/python3.10/dist-packages/hydra/core/utils.py", line 260, in return_value
raise self._return_value
File "/usr/local/lib/python3.10/dist-packages/hydra/core/utils.py", line 186, in run_job
ret.return_value = task_function(task_cfg)
File "/workspace/DreamFusion/main.py", line 60, in main
trainer.fit(model, datamodule=dm)
File "/usr/local/lib/python3.10/dist-packages/lightning/pytorch/trainer/trainer.py", line 532, in fit
model = _maybe_unwrap_optimized(model)
File "/usr/local/lib/python3.10/dist-packages/lightning/pytorch/utilities/compile.py", line 111, in _maybe_unwrap_optimized
raise TypeError(
TypeError: `model` must be a `LightningModule` or `torch._dynamo.OptimizedModule`, got `DreamFusion`
I have tried running it using the nvcr.io/nvidia/nemo:dev, but still got the same errors.