TencentARC

/

Pixal3D

like 204

Follow

ARC Lab, Tencent PCG 967

[SPARSE] Conv backend: flex_gemm; Attention backend: flash_attn
============================================================
[Diagnostics] PyTorch version: 2.11.0+cu128
[Diagnostics] CUDA available: True
[Diagnostics] CUDA version: 12.8
[Diagnostics] cuDNN version: 91900
[Diagnostics] GPU 0: NVIDIA RTX PRO 6000 Blackwell Workstation Edition, sm_120, 95.6 GB
[Diagnostics] nvidia-smi: NVIDIA RTX PRO 6000 Blackwell Workstation Edition, 12.0, 97887 MiB
============================================================
[Pipeline] Loading from TencentARC/Pixal3D...
[ATTENTION] Using backend: flash_attn
/opt/miniconda3/envs/pixal3d/lib/python3.12/site-packages/timm/models/layers/__init__.py:48: FutureWarning: Importing from timm.models.layers is deprecated, please import via timm.layers
  warnings.warn(f"Importing from {__name__} is deprecated, please import via timm.layers", FutureWarning)
/opt/miniconda3/envs/pixal3d/lib/python3.12/site-packages/timm/models/registry.py:4: FutureWarning: Importing from timm.models.registry is deprecated, please import via timm.models
  warnings.warn(f"Importing from {__name__} is deprecated, please import via timm.models", FutureWarning)
[ImageCond] Building DinoV3ProjFeatureExtractor models...
Traceback (most recent call last):
  File "/root/Develop/Pixal3D/app.py", line 557, in <module>
    init_models()
  File "/root/Develop/Pixal3D/app.py", line 175, in init_models
    pipeline.cuda()
  File "/root/Develop/Pixal3D/pixal3d/pipelines/base.py", line 69, in cuda
    self.to(torch.device("cuda"))
  File "/root/Develop/Pixal3D/pixal3d/pipelines/pixal3d_image_to_3d.py", line 140, in to
    super().to(device)
  File "/root/Develop/Pixal3D/pixal3d/pipelines/base.py", line 66, in to
    model.to(device)
  File "/opt/miniconda3/envs/pixal3d/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1384, in to
    return self._apply(convert)
           ^^^^^^^^^^^^^^^^^^^^
  File "/opt/miniconda3/envs/pixal3d/lib/python3.12/site-packages/torch/nn/modules/module.py", line 934, in _apply
    module._apply(fn)
  File "/opt/miniconda3/envs/pixal3d/lib/python3.12/site-packages/torch/nn/modules/module.py", line 965, in _apply
    param_applied = fn(param)
                    ^^^^^^^^^
  File "/opt/miniconda3/envs/pixal3d/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1370, in convert
    return t.to(
           ^^^^^
RuntimeError: CUDA driver error: unknown error