hzxie / CityDreamer

The official implementation of "CityDreamer: Compositional Generative Model of Unbounded 3D Cities". (Xie et al., CVPR 2024)

Home Page:https://haozhexie.com/project/city-dreamer

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

RuntimeError: No CUDA GPUs are available

dastan1910 opened this issue · comments

RuntimeError: No CUDA GPUs are available

[INFO] 2024-04-02 14:55:42,701 192.168.31.42 - - [02/Apr/2024 14:55:42] "GET /image/e5fb56dc-5324-4875-b320-c69c81d52e0c.png/normalize.action HTTP/1.1" 200 -
[INFO] 2024-04-02 14:55:42,707 192.168.31.42 - - [02/Apr/2024 14:55:42] "GET /image/5dfef487-02fd-4725-a58e-1a4bae428e6c.png HTTP/1.1" 200 -
Rendering seg volume:   0%|                                                                                                                                                                 | 0/20 [00:00<?, ?it/s]
[INFO] 2024-04-02 14:55:44,741 192.168.31.42 - - [02/Apr/2024 14:55:44] "POST /trajectory/preview.action HTTP/1.1" 500 -
Traceback (most recent call last):
  File "/home/dastan/.virtualenvs/citydreamer/lib/python3.10/site-packages/flask/app.py", line 1488, in __call__
    return self.wsgi_app(environ, start_response)
  File "/home/dastan/.virtualenvs/citydreamer/lib/python3.10/site-packages/flask/app.py", line 1466, in wsgi_app
    response = self.handle_exception(e)
  File "/home/dastan/.virtualenvs/citydreamer/lib/python3.10/site-packages/flask/app.py", line 1463, in wsgi_app
    response = self.full_dispatch_request()
  File "/home/dastan/.virtualenvs/citydreamer/lib/python3.10/site-packages/flask/app.py", line 872, in full_dispatch_request
    rv = self.handle_user_exception(e)
  File "/home/dastan/.virtualenvs/citydreamer/lib/python3.10/site-packages/flask/app.py", line 870, in full_dispatch_request
    rv = self.dispatch_request()
  File "/home/dastan/.virtualenvs/citydreamer/lib/python3.10/site-packages/flask/app.py", line 855, in dispatch_request
    return self.ensure_sync(self.view_functions[rule.endpoint])(**view_args)  # type: ignore[no-any-return]
  File "/home/dastan/kas/projects/city-dreamer/demo/run.py", line 164, in get_trajectory_preview
    frames = get_seg_volume_rendering(hf_filepath, seg_filepath, trajectory)
  File "/home/dastan/kas/projects/city-dreamer/demo/run.py", line 204, in get_seg_volume_rendering
    seg_volume = inference.get_seg_volume(part_hf, part_seg)
  File "/home/dastan/kas/projects/city-dreamer/demo/../scripts/inference.py", line 376, in get_seg_volume
    return scripts.dataset_generator.get_seg_volume(part_seg, part_hf)
  File "/home/dastan/kas/projects/city-dreamer/demo/../scripts/dataset_generator.py", line 84, in get_seg_volume
    torch.from_numpy(part_seg_map[None, None, ...]).cuda(),
  File "/home/dastan/.virtualenvs/citydreamer/lib/python3.10/site-packages/torch/cuda/__init__.py", line 229, in _lazy_init
    torch._C._cuda_init()
RuntimeError: No CUDA GPUs are available