Skip to content

RuntimeError: No CUDA GPUs are available #44

@lambda7xx

Description

@lambda7xx

I run the python3 offline_inference.py, then I meet this error.

INFO 05-19 10:39:22 base_llm_engine.py:190] Initializing workers with distributed init method: tcp://172.17.0.4:36106
len(self.workers):[Actor(RayWorker, b387e8cfcb6791f20607568201000000)]
rank:0 and worker:Actor(RayWorker, b387e8cfcb6791f20607568201000000) and local_rank:0
Traceback (most recent call last):
  File "/root/autodl-tmp/vattention/sarathi-lean/examples/offline_inference.py", line 26, in <module>
    llm_engine = LLMEngine.from_engine_args(
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/sarathi-0.1.7-py3.10-linux-x86_64.egg/sarathi/engine/llm_engine.py", line 17, in from_engine_args
    engine = BaseLLMEngine(*engine_configs)
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/sarathi-0.1.7-py3.10-linux-x86_64.egg/sarathi/engine/base_llm_engine.py", line 111, in __init__
    self._init_workers_ray()
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/sarathi-0.1.7-py3.10-linux-x86_64.egg/sarathi/engine/base_llm_engine.py", line 221, in _init_workers_ray
    self._run_workers(
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/sarathi-0.1.7-py3.10-linux-x86_64.egg/sarathi/engine/base_llm_engine.py", line 440, in _run_workers
    all_outputs = ray.get(all_outputs)
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/ray/_private/auto_init_hook.py", line 18, in auto_init_wrapper
    return fn(*args, **kwargs)
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/ray/_private/client_mode_hook.py", line 103, in wrapper
    return func(*args, **kwargs)
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/ray/_private/worker.py", line 2540, in get
    raise value.as_instanceof_cause()
ray.exceptions.RayTaskError(RuntimeError): �[36mray::RayWorker.execute_method()�[39m (pid=25233, ip=172.17.0.4, actor_id=b387e8cfcb6791f20607568201000000, repr=<sarathi.engine.ray_utils.RayWorker object at 0x7fed2d109300>)
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/sarathi-0.1.7-py3.10-linux-x86_64.egg/sarathi/engine/ray_utils.py", line 32, in execute_method
    return executor(*args, **kwargs)
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 116, in decorate_context
    return func(*args, **kwargs)
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/sarathi-0.1.7-py3.10-linux-x86_64.egg/sarathi/utils/threading_utils.py", line 17, in synced_method
    return method(self, *args, **kwargs)
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/sarathi-0.1.7-py3.10-linux-x86_64.egg/sarathi/worker/base_worker.py", line 99, in init_model
    torch.cuda.set_device(self.device)
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/torch/cuda/__init__.py", line 420, in set_device
    torch._C._cuda_setDevice(device)
  File "/root/miniconda3/envs/vattn/lib/python3.10/site-packages/torch/cuda/__init__.py", line 314, in _lazy_init
    torch._C._cuda_init()
RuntimeError: No CUDA GPUs are available

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions