-
Notifications
You must be signed in to change notification settings - Fork 33
Description
In Ubuntu2404,python3.10, torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0
IT seems an enviroment problem,i wonder what system ,gcc are tested ,and how to fix this ?
File "/home/code/wwg/AnySplat/inference.py", line 39, in
main()
File "/home/code/wwg/AnySplat/inference.py", line 30, in main
gaussians, pred_context_pose = model.inference((images+1)*0.5)
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "/home/code/wwg/AnySplat/src/model/model/anysplat.py", line 101, in inference
encoder_output = self.encoder(context_image, global_step=0, visualization_dump=None)
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/home/code/wwg/AnySplat/src/model/encoder/anysplat.py", line 405, in forward
aggregated_tokens_list, patch_start_idx = self.aggregator(
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/home/code/wwg/AnySplat/src/model/encoder/vggt/models/aggregator.py", line 212, in forward
patch_tokens = self.patch_embed(images)
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/home/code/wwg/AnySplat/src/model/encoder/vggt/layers/vision_transformer.py", line 337, in forward
ret = self.forward_features(*args, **kwargs)
File "/home/code/wwg/AnySplat/src/model/encoder/vggt/layers/vision_transformer.py", line 267, in forward_features
x = self.prepare_tokens_with_masks(x, masks)
File "/home/code/wwg/AnySplat/src/model/encoder/vggt/layers/vision_transformer.py", line 220, in prepare_tokens_with_masks
x = self.patch_embed(x)
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/home/code/wwg/AnySplat/src/model/encoder/vggt/layers/patch_embed.py", line 75, in forward
x = self.proj(x) # B C H W
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/nn/modules/conv.py", line 460, in forward
return self._conv_forward(input, self.weight, self.bias)
File "/home/data/miniconda3/envs/python10/lib/python3.10/site-packages/torch/nn/modules/conv.py", line 456, in _conv_forward
return F.conv2d(input, weight, bias, self.stride,
RuntimeError: GET was unable to find an engine to execute this computation