Oleg Shulyakov commited on
Commit
a0296c1
·
1 Parent(s): e576d87

Dockerfile CUDA

Browse files
Files changed (1) hide show
  1. .devops/cuda.Dockerfile +28 -0
.devops/cuda.Dockerfile ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 \
9
+ PATH=${PATH}:/app \
10
+ PATH=${PATH}:/usr/local/nvidia/bin
11
+
12
+ WORKDIR ${HOME}/app
13
+
14
+ COPY --chown=1000 . ${HOME}/app
15
+
16
+ RUN pip install --no-cache-dir -r requirements.txt
17
+
18
+ ENV PYTHONPATH=${PYTHONPATH}:${HOME}/.local/bin \
19
+ LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/app \
20
+ LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib64 \
21
+ HF_HUB_ENABLE_HF_TRANSFER=1 \
22
+ GRADIO_ALLOW_FLAGGING=never \
23
+ GRADIO_NUM_PORTS=1 \
24
+ GRADIO_SERVER_NAME=0.0.0.0 \
25
+ GRADIO_THEME=huggingface \
26
+ SYSTEM=spaces
27
+
28
+ ENTRYPOINT ["/bin/bash", "start.sh"]