Oleg Shulyakov commited on
Commit
0a7c60c
·
1 Parent(s): c09ee0a

docker: CUDA

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -1,11 +1,11 @@
1
- FROM ghcr.io/ggml-org/llama.cpp:full
2
 
3
  # Check if user with UID 1000 exists, if not create it
4
  RUN id -u 1000 &>/dev/null || useradd -m -u 1000 user
5
  USER 1000
6
 
7
  ENV HOME=/home/user \
8
- PATH=${PATH}:/home/user/.local/bin:/app
9
 
10
  WORKDIR ${HOME}/app
11
 
@@ -14,7 +14,7 @@ COPY --chown=1000 requirements.txt ${HOME}/app
14
  RUN pip install --no-cache-dir -r requirements.txt
15
 
16
  ENV PYTHONPATH=${PYTHONPATH}:${HOME}/.local/bin \
17
- LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/app \
18
  HF_HUB_ENABLE_HF_TRANSFER=1 \
19
  GRADIO_ALLOW_FLAGGING=never \
20
  GRADIO_NUM_PORTS=1 \
 
1
+ FROM ghcr.io/ggml-org/llama.cpp:full-cuda
2
 
3
  # Check if user with UID 1000 exists, if not create it
4
  RUN id -u 1000 &>/dev/null || useradd -m -u 1000 user
5
  USER 1000
6
 
7
  ENV HOME=/home/user \
8
+ PATH=${PATH}:/home/user/.local/bin:/usr/local/nvidia/bin:/app
9
 
10
  WORKDIR ${HOME}/app
11
 
 
14
  RUN pip install --no-cache-dir -r requirements.txt
15
 
16
  ENV PYTHONPATH=${PYTHONPATH}:${HOME}/.local/bin \
17
+ LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib64:/app \
18
  HF_HUB_ENABLE_HF_TRANSFER=1 \
19
  GRADIO_ALLOW_FLAGGING=never \
20
  GRADIO_NUM_PORTS=1 \