abetlen / abetlen/llama-cpp-python

Metal installation documentation

Open
#1,968 0 comments 3 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
10.6k
Forks
1.4k
PR merge metrics
PR metrics pending

Description

I tried setting up `llama-cpp-python` in the current version `0.3.7` on my MacBook M4 Pro.
In the first step I only installed via `pip install llama-cpp-python --no-cache-dir` without specifiying the environment variable for Metal backend support.
I set the `n_gpu_layers` to `-1` to fully use the GPU.

The interesting thing is the GPU was used even without having to install the Metal backend support as stated in the current documentation. I double checked this with a fresh start and explicitely setting the `CMAKE_ARGS` env variable and did not see and difference in terms of performance or GPU usage.

This is pretty handy, because when not using `pip` for dependency management (e.g. `poetry`) passing the environment variable did not work on my side.

Maybe the documentation should be updated to state that the env arguments are no longer required? This would also reflect the documentation in https://github.com/ggml-org/llama.cpp/blob/master/docs/build.md#metal-build where it states that: "On MacOS, Metal is enabled by default"

This would also mean, that custom pre-built wheels are no longer required as well.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.