Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

GPU support may not be enabled #62

Closed
svanschalkwyk opened this issue Oct 21, 2023 · 5 comments
Closed

GPU support may not be enabled #62

svanschalkwyk opened this issue Oct 21, 2023 · 5 comments

Comments

@svanschalkwyk
Copy link

"routes.go:634: Warning: GPU support may not enabled, check you have installed install GPU drivers: nvidia-smi command failed"
NVIDIA-SMI 535.86.10 Driver Version: 535.86.10 CUDA Version: 12.2
5.15.0-86-generic #96-Ubuntu
docker run --rm --gpus all nvidia/cuda:11.6.2-base-ubuntu20.04 nvidia-smi:
NVIDIA-SMI 535.86.10 Driver Version: 535.86.10 CUDA Version: 12.2

@svanschalkwyk
Copy link
Author

llama.go:333: skipping accelerated runner because num_gpu=0
genai-stack-llm-1 | 2023/10/21 03:52:30 llama.go:356: starting llama runner
genai-stack-llm-1 | 2023/10/21 03:52:30 llama.go:408: waiting for llama runner to start responding
genai-stack-llm-1 | [GIN] 2023/10/21 - 03:52:30 | 500 | 873.831145ms | 172.20.0.5 | POST "/api/generate"
genai-stack-llm-1 | 2023/10/21 03:52:30 llama.go:373: error starting llama runner: llama runner process has terminated
genai-stack-llm-1 | 2023/10/21 03:52:30 llama.go:438: llama runner stopped with error: signal: illegal instruction (core dumped)

@jexp
Copy link
Collaborator

jexp commented Oct 21, 2023

Is that on linux? Can you share more of your OS and machine specs.
@mchiang0610

@matthieuml
Copy link
Contributor

matthieuml commented Oct 21, 2023

If you are using the Linux profile, you should add this to the llm service:

deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

And ensure that you have nvidia-container-toolkit installed

@svanschalkwyk
Copy link
Author

svanschalkwyk commented Oct 22, 2023 via email

@jexp
Copy link
Collaborator

jexp commented Nov 3, 2023

Added it to the readme.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants