ModuleNotFoundError: No module named 'Swin3D.sparse_dl.attn_cuda'

[jiachen0212]

thanks for your work~!
but when i run:

import torch
from Swin3D.models import Swin3DUNet
model = Swin3DUNet(depths, channels, num_heads,
window_sizes, quant_size, up_k=up_k,
drop_path_rate=drop_path_rate, num_classes=num_classes,
num_layers=num_layers, stem_transformer=stem_transformer,
upsample=upsample, first_down_stride=down_stride,
knn_down=knn_down, in_channels=in_channels,
cRSE='XYZ_RGB_NORM', fp16_mode=1)
model.load_pretrained_model(ckpt_path)

i got this bug: ModuleNotFoundError: No module named 'Swin3D.sparse_dl.attn_cuda'
emmm, what can i do???

[Abokhatwa]

Were you able to solve this problem ? because I am facing the same issue.

[Yukichiii]

Could you please provide your python version?
And did you run the command python setup.py install?

[dang-c]

Could you please provide your python version? And did you run the command python setup.py install?

I also encountered this problem. Here are my environment information
python:3.10
pytorch:2.5.1+
Ubuntu:20.04
CUDA:12.1
I have run python setup.py install and it shows that the installation is successful.
Here is my error message:
import Swin3D.sparse_dl.attn_cuda as attn_module
ModuleNotFoundError: No module named 'Swin3D.sparse_dl.attn_cuda'
What should I do? I look forward to your reply

[dang-c]

Could you please provide your python version? And did you run the command python setup.py install?

I also encountered this problem. Here are my environment information python:3.10 pytorch:2.5.1+ Ubuntu:20.04 CUDA:12.1 I have run python setup.py install and it shows that the installation is successful. Here is my error message: import Swin3D.sparse_dl.attn_cuda as attn_module ModuleNotFoundError: No module named 'Swin3D.sparse_dl.attn_cuda' What should I do? I look forward to your reply

I solved this problem by this command, hope it helps you
python setup.py build_ext --inplace