AiuniAI / Unique3D

[NeurIPS 2024] Unique3D: High-Quality and Efficient 3D Mesh Generation from a Single Image
https://wukailu.github.io/Unique3D/
MIT License
3.1k stars 248 forks source link

got RuntimeError: Not compiled with CUDA support #108

Open zxzAndyMAC opened 1 month ago

zxzAndyMAC commented 1 month ago

when I click "Generate 3D" button, I got error on runtime: Traceback (most recent call last): File "/home/fozmo/anaconda3/envs/unique3d/lib/python3.11/site-packages/gradio/queueing.py", line 536, in process_events response = await route_utils.call_process_api( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/anaconda3/envs/unique3d/lib/python3.11/site-packages/gradio/route_utils.py", line 322, in call_process_api output = await app.get_blocks().process_api( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/anaconda3/envs/unique3d/lib/python3.11/site-packages/gradio/blocks.py", line 1935, in process_api result = await self.call_function( ^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/anaconda3/envs/unique3d/lib/python3.11/site-packages/gradio/blocks.py", line 1520, in call_function prediction = await anyio.to_thread.run_sync( # type: ignore ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/anaconda3/envs/unique3d/lib/python3.11/site-packages/anyio/to_thread.py", line 56, in run_sync return await get_async_backend().run_sync_in_worker_thread( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/anaconda3/envs/unique3d/lib/python3.11/site-packages/anyio/_backends/_asyncio.py", line 2405, in run_sync_in_worker_thread return await future ^^^^^^^^^^^^ File "/home/fozmo/anaconda3/envs/unique3d/lib/python3.11/site-packages/anyio/_backends/_asyncio.py", line 914, in run result = context.run(func, args) ^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/anaconda3/envs/unique3d/lib/python3.11/site-packages/gradio/utils.py", line 826, in wrapper response = f(args, kwargs) ^^^^^^^^^^^^^^^^^^ File "/home/fozmo/Dev/new3d/Unique3D/app/gradio_3dgen.py", line 21, in generate3dv2 new_meshes = geo_reconstruct(rgb_pils, None, front_pil, do_refine=do_refine, predict_normal=True, expansion_weight=expansion_weight, init_type=init_type) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/Dev/new3d/Unique3D/scripts/multiview_inference.py", line 92, in geo_reconstruct vertices, faces = reconstruct_stage1(normal_stg1, steps=200, vertices=vertices, faces=faces, start_edge_len=0.1, end_edge_len=0.02, gain=0.05, return_mesh=False, loss_expansion_weight=expansion_weight) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/Dev/new3d/Unique3D/mesh_reconstruction/recon.py", line 52, in reconstruct_stage1 vertices,faces = opt.remesh(poisson=False) ^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/Dev/new3d/Unique3D/mesh_reconstruction/opt.py", line 185, in remesh self._vertices_etc,self._faces = remesh(self._vertices_etc,self._faces,min_edge_len,max_edge_len,flip, max_vertices=1e6) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/anaconda3/envs/unique3d/lib/python3.11/site-packages/torch/utils/_contextlib.py", line 116, in decorate_context return func(*args, *kwargs) ^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/Dev/new3d/Unique3D/mesh_reconstruction/opt.py", line 33, in remesh vertices_etc,faces = collapse_edges(vertices_etc,faces,edges,priority) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/Dev/new3d/Unique3D/mesh_reconstruction/remesh.py", line 219, in collapse_edges torch_scatter.scatter_max(src=edge_rank[:,None].expand(-1,2).reshape(-1),index=edges.reshape(-1),dim=0,out=vert_rank) File "/home/fozmo/anaconda3/envs/unique3d/lib/python3.11/site-packages/torch_scatter/scatter.py", line 72, in scatter_max return torch.ops.torch_scatter.scatter_max(src, index, dim, out, dim_size) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/fozmo/anaconda3/envs/unique3d/lib/python3.11/site-packages/torch/ops.py", line 1061, in call return self._op(args, (kwargs or {})) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: Not compiled with CUDA support