natten安装不成功，使用whl安装后，后面也会报错

[hhhhhhao-1]

(emrdm) root@autodl-container-9b4c40bda6-879805ae:~/autodl-tmp/EMRDM# pip install natten==0.17.1
Looking in indexes: http://mirrors.aliyun.com/pypi/simple
Collecting natten==0.17.1
Using cached http://mirrors.aliyun.com/pypi/packages/06/64/c5a2773705d2cd09284d03368ac4a8fc85fb9da02514fc3aefa43692a775/natten-0.17.1.tar.gz (10.8 MB)
Installing build dependencies ... done
Getting requirements to build wheel ... error
error: subprocess-exited-with-error

× Getting requirements to build wheel did not run successfully.
│ exit code: 1
╰─> [17 lines of output]
Traceback (most recent call last):
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 389, in
main()
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 373, in main
json_out["return_val"] = hook(**hook_input["kwargs"])
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 143, in get_requires_for_build_wheel
return hook(config_settings)
File "/tmp/pip-build-env-6sfqb3ma/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 331, in get_requires_for_build_wheel
return self._get_build_requires(config_settings, requirements=[])
File "/tmp/pip-build-env-6sfqb3ma/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 301, in _get_build_requires
self.run_setup()
File "/tmp/pip-build-env-6sfqb3ma/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 512, in run_setup
super().run_setup(setup_script=setup_script)
File "/tmp/pip-build-env-6sfqb3ma/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 317, in run_setup
exec(code, locals())
File "", line 34, in
ModuleNotFoundError: No module named 'torch'
[end of output]

note: This error originates from a subprocess, and is likely not a problem with pip.
ERROR: Failed to build 'natten' when getting requirements to build wheel

[hhhhhhao-1]

下载whl文件安装natten的话，会报错，如下：
trainer:
devices: auto
num_sanity_val_steps: 0
benchmark: true
accumulate_grad_batches: 1
check_val_every_n_epoch: 1
max_epochs: 500

Testing DataLoader 0: 0%| | 0/34 [00:00<?, ?it/s]Plotting: Switched to EMA weights
Plotting: Restored training weights
Summoning checkpoint.
Traceback (most recent call last):
File "/root/autodl-tmp/EMRDM/main.py", line 939, in
trainer.test(model, data)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pytorch_lightning/trainer/trainer.py", line 753, in test
return call._call_and_handle_interrupt(
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pytorch_lightning/trainer/call.py", line 43, in _call_and_handle_interrupt
return trainer.strategy.launcher.launch(trainer_fn, *args, trainer=trainer, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pytorch_lightning/strategies/launchers/subprocess_script.py", line 105, in launch
return function(*args, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pytorch_lightning/trainer/trainer.py", line 793, in _test_impl
results = self._run(model, ckpt_path=ckpt_path)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pytorch_lightning/trainer/trainer.py", line 986, in _run
results = self._run_stage()
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pytorch_lightning/trainer/trainer.py", line 1023, in _run_stage
return self._evaluation_loop.run()
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pytorch_lightning/loops/utilities.py", line 182, in _decorator
return loop_run(self, *args, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pytorch_lightning/loops/evaluation_loop.py", line 135, in run
self._evaluation_step(batch, batch_idx, dataloader_idx, dataloader_iter)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pytorch_lightning/loops/evaluation_loop.py", line 396, in _evaluation_step
output = call._call_strategy_hook(trainer, hook_name, *step_args)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pytorch_lightning/trainer/call.py", line 311, in _call_strategy_hook
output = fn(*args, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pytorch_lightning/strategies/strategy.py", line 424, in test_step
return self.lightning_module.test_step(*args, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "/root/autodl-tmp/EMRDM/sgm/models/diffusion.py", line 683, in test_step
self.shared_test_step(batch=batch)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "/root/autodl-tmp/EMRDM/sgm/models/diffusion.py", line 633, in shared_test_step
samples, _ = self.sample(
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "/root/autodl-tmp/EMRDM/sgm/models/diffusion.py", line 451, in sample
samples = self.sampler(denoiser, randn, mu, cond, uc=uc, return_intermediate=return_intermediate, return_denoised=return_denoised)
File "/root/autodl-tmp/EMRDM/sgm/modules/diffusionmodules/sampling.py", line 496, in call
x, denoised = self.sampler_step(
File "/root/autodl-tmp/EMRDM/sgm/modules/diffusionmodules/sampling.py", line 466, in sampler_step
denoised = self.denoise(x, denoiser, sigma_hat, cond, st_hat, uc)
File "/root/autodl-tmp/EMRDM/sgm/modules/diffusionmodules/sampling.py", line 399, in denoise
denoised = denoiser(*self.guider.prepare_inputs(x, sigma, cond, uc),st=st)
File "/root/autodl-tmp/EMRDM/sgm/models/diffusion.py", line 445, in
denoiser = lambda input, sigma, c, st: self.denoiser(
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/root/autodl-tmp/EMRDM/sgm/modules/diffusionmodules/denoiser.py", line 97, in forward
network(input * c_in, c_noise, cond, **additional_model_inputs) * c_out
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/root/autodl-tmp/EMRDM/sgm/modules/diffusionmodules/wrappers.py", line 43, in forward
return self.diffusion_model(
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/root/autodl-tmp/EMRDM/sgm/modules/diffusionmodules/k_diffusion/image_transformer.py", line 1186, in forward
x = down_level(x, pos, cond)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/root/autodl-tmp/EMRDM/sgm/modules/diffusionmodules/k_diffusion/image_transformer.py", line 531, in forward
x = layer(x, *args, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/root/autodl-tmp/EMRDM/sgm/modules/diffusionmodules/k_diffusion/image_transformer.py", line 500, in forward
x = checkpoint(self.self_attn, x, pos, cond)
File "/root/autodl-tmp/EMRDM/sgm/modules/diffusionmodules/k_diffusion/image_transformer.py", line 44, in checkpoint
return function(*args, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/root/autodl-tmp/EMRDM/sgm/modules/diffusionmodules/k_diffusion/image_transformer.py", line 405, in forward
raise ModuleNotFoundError("natten is required for neighborhood attention")
ModuleNotFoundError: natten is required for neighborhood attention

[hhhhhhao-1]

好吧，要用cu121版本，已经可以了这一步

[Ly403]

(emrdm) root@autodl-container-9b4c40bda6-879805ae:~/autodl-tmp/EMRDM# pip install natten==0.17.1 Looking in indexes: http://mirrors.aliyun.com/pypi/simple Collecting natten==0.17.1 Using cached http://mirrors.aliyun.com/pypi/packages/06/64/c5a2773705d2cd09284d03368ac4a8fc85fb9da02514fc3aefa43692a775/natten-0.17.1.tar.gz (10.8 MB) Installing build dependencies ... done Getting requirements to build wheel ... error error: subprocess-exited-with-error

× Getting requirements to build wheel did not run successfully. │ exit code: 1 ╰─> [17 lines of output] Traceback (most recent call last): File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 389, in main() File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 373, in main json_out["return_val"] = hook(**hook_input["kwargs"]) File "/root/miniconda3/envs/emrdm/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 143, in get_requires_for_build_wheel return hook(config_settings) File "/tmp/pip-build-env-6sfqb3ma/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 331, in get_requires_for_build_wheel return self._get_build_requires(config_settings, requirements=[]) File "/tmp/pip-build-env-6sfqb3ma/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 301, in _get_build_requires self.run_setup() File "/tmp/pip-build-env-6sfqb3ma/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 512, in run_setup super().run_setup(setup_script=setup_script) File "/tmp/pip-build-env-6sfqb3ma/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 317, in run_setup exec(code, locals()) File "", line 34, in ModuleNotFoundError: No module named 'torch' [end of output]

note: This error originates from a subprocess, and is likely not a problem with pip. ERROR: Failed to build 'natten' when getting requirements to build wheel

这个不是因为没有装torch吗

[Ly403]

好吧，要用cu121版本，已经可以了这一步

wheel是编译好的版本，肯定是要和torch和cuda版本对齐的

[hhhhhhao-1]

您好，好像这里需要修改一下

[Ly403]

您好，好像这里需要修改一下

您好，这个是因为我们的Dataset实际上用的是UnCRtainTS实现的代码，因为该仓库实现的Dataset功能要比原始的SEN12MS-CR-TS实现的更丰富一点，接口也更友好一些。而且，UnCRtainTS和SEN12MS-CR-TS是同一作者团队的工作。