matthoffner commited on
Commit
707c746
·
verified ·
1 Parent(s): 2c4789e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -7
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM nvidia/cuda:12.2.0-devel-ubuntu22.04
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
 
@@ -16,11 +16,6 @@ RUN apt update && \
16
  libblas-dev && \
17
  apt clean && \
18
  rm -rf /var/lib/apt/lists/*
19
-
20
- # Setting up CUDA environment variables (this may not be necessary since you're using the official nvidia/cuda image, but it's good to be explicit)
21
- ENV PATH="/usr/local/cuda/bin:$PATH" \
22
- LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH" \
23
- CUDA_HOME="/usr/local/cuda"
24
 
25
  WORKDIR /app
26
 
@@ -33,7 +28,7 @@ RUN git clone https://github.com/ggerganov/llama.cpp.git && \
33
  cd llama.cpp && \
34
  git submodule init && \
35
  git submodule update && \
36
- make LLAMA_CUBLAS=1
37
 
38
  # Create a non-root user for security reasons
39
  RUN useradd -m -u 1000 user && \
 
1
+ FROM ubuntu:22.04
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
 
 
16
  libblas-dev && \
17
  apt clean && \
18
  rm -rf /var/lib/apt/lists/*
 
 
 
 
 
19
 
20
  WORKDIR /app
21
 
 
28
  cd llama.cpp && \
29
  git submodule init && \
30
  git submodule update && \
31
+ make
32
 
33
  # Create a non-root user for security reasons
34
  RUN useradd -m -u 1000 user && \