3DIAS_PyTorch icon indicating copy to clipboard operation
3DIAS_PyTorch copied to clipboard

AttributeError: module 'torchmcubes_module' has no attribute 'mcubes_cuda'

Open CAOXINGWEN opened this issue 2 years ago • 1 comments

Hey, thank you for your excellent work. I encountered a problem while running the demo. py file. Can you help me take a look

pythone demo.py --device "0" --inputimg "./input/2.png" --config "./weights/single/chair/config.json" --resume "./weights/single/chair/checkpoint-epoch546.pth" load model Traceback (most recent call last): File "demo.py", line 77, in main(config) File "demo.py", line 36, in main meshlist, total_mesh, _ = generate_mesh(polycoeff, A_10x10) File "/media/caoxingwen/work/3DIAS/utils/generate_figure.py", line 36, in generate_mesh verts0, faces0 = marching_cubes(func_val, 0.0) File "/home/caoxingwen/anaconda3/envs/3dais/lib/python3.8/site-packages/torchmcubes/init.py", line 12, in marching_cubes return mc.mcubes_cuda(vol, thresh) AttributeError: module 'torchmcubes_module' has no attribute 'mcubes_cuda'

CAOXINGWEN avatar Apr 21 '23 09:04 CAOXINGWEN

Hi, torchmcubes library has two different versions: pip version and git repo verison. We used git repository version which supports cuda calculation As noticed in README.md, see here

robot0321 avatar May 19 '23 06:05 robot0321