2023-05-31 16:00:31 +00:00
# Install Docker Server
**Note #1: ** This was tested with Docker running on Linux. If you can get it working on Windows or MacOS, please update this `README.md` with a PR!
[Install Docker Engine ](https://docs.docker.com/engine/install )
**Note #2: ** NVidia GPU CuBLAS support requires a NVidia GPU with sufficient VRAM (approximately as much as the size above) and Docker NVidia support (see [container-toolkit/install-guide ](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html ))
2023-05-31 15:16:32 +00:00
# Simple Dockerfiles for building the llama-cpp-python server with external model bin files
- `./openblas_simple/Dockerfile` - a simple Dockerfile for non-GPU OpenBLAS, where the model is located outside the Docker image
- `cd ./openblas_simple`
- `docker build -t openblas_simple .`
- `docker run -e USE_MLOCK=0 -e MODEL=/var/model/<model-path> -v <model-root-path>:/var/model -t openblas_simple`
where `<model-root-path>/<model-path>` is the full path to the model file on the Docker host system.
- `./cuda_simple/Dockerfile` - a simple Dockerfile for CUDA accelerated CuBLAS, where the model is located outside the Docker image
- `cd ./cuda_simple`
- `docker build -t cuda_simple .`
- `docker run -e USE_MLOCK=0 -e MODEL=/var/model/<model-path> -v <model-root-path>:/var/model -t cuda_simple`
where `<model-root-path>/<model-path>` is the full path to the model file on the Docker host system.
2023-06-02 10:44:52 +00:00
# "Open-Llama-in-a-box" - Download a MIT licensed Open Llama model and install into a Docker image that runs an OpenBLAS-enabled llama-cpp-python server
```
$ cd ./open_llama
./build.sh
./start.sh
```
# Manually choose your own Llama model from Hugging Face
- `python3 ./hug_model.py -a TheBloke -t llama`
2023-05-31 16:00:31 +00:00
- You should now have a model in the current directory and `model.bin` symlinked to it for the subsequent Docker build and copy step. e.g.
2023-05-23 19:26:40 +00:00
```
docker $ ls -lh *.bin
2023-05-31 16:00:31 +00:00
-rw-rw-r-- 1 user user 4.8G May 23 18:30 < downloaded-model-file > q5_1.bin
lrwxrwxrwx 1 user user 24 May 23 18:30 model.bin -> < downloaded-model-file > q5_1.bin
2023-05-23 19:26:40 +00:00
```
2023-05-23 19:50:39 +00:00
**Note #1: ** Make sure you have enough disk space to download the model. As the model is then copied into the image you will need at least
2023-05-23 19:26:40 +00:00
**TWICE** as much disk space as the size of the model:
| Model | Quantized size |
|------:|----------------:|
2023-06-02 08:48:54 +00:00
| 3B | 3 GB |
2023-05-23 19:26:40 +00:00
| 7B | 5 GB |
| 13B | 10 GB |
2023-06-02 08:48:54 +00:00
| 33B | 25 GB |
2023-05-23 19:26:40 +00:00
| 65B | 50 GB |
2023-05-23 19:36:21 +00:00
**Note #2: ** If you want to pass or tune additional parameters, customise `./start_server.sh` before running `docker build ...`
2023-05-23 19:26:40 +00:00
2023-05-31 16:00:31 +00:00
## Use OpenBLAS
2023-05-23 19:50:39 +00:00
Use if you don't have a NVidia GPU. Defaults to `python:3-slim-bullseye` Docker base image and OpenBLAS:
2023-05-31 16:00:31 +00:00
### Build:
`docker build -t openblas .`
### Run:
2023-05-23 19:26:40 +00:00
`docker run --cap-add SYS_RESOURCE -t openblas`
2023-05-31 16:00:31 +00:00
## Use CuBLAS
### Build:
2023-05-23 19:50:39 +00:00
`docker build --build-arg IMAGE=nvidia/cuda:12.1.1-devel-ubuntu22.04 -t cublas .`
2023-05-31 16:00:31 +00:00
### Run:
2023-05-23 19:26:40 +00:00
`docker run --cap-add SYS_RESOURCE -t cublas`