matthoffner commited on
Commit
98cbe10
·
1 Parent(s): 44cac5d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -5
Dockerfile CHANGED
@@ -3,11 +3,11 @@ FROM nvidia/cuda:12.2.0-devel-ubuntu20.04
3
  ENV DEBIAN_FRONTEND=noninteractive
4
 
5
  # Add NVIDIA CUDA Repository
6
- RUN apt update && apt install -y wget gnupg && \
7
  wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin && \
8
- sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \
9
- sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub && \
10
- sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
11
 
12
  # Install dependencies
13
  RUN apt update && \
@@ -15,7 +15,6 @@ RUN apt update && \
15
  apt install -y libcublas-dev && \
16
  apt clean && rm -rf /var/lib/apt/lists/*
17
 
18
-
19
  WORKDIR /app
20
 
21
  # Download ggml and mmproj models from HuggingFace
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
 
5
  # Add NVIDIA CUDA Repository
6
+ RUN apt update && apt install -y wget gnupg software-properties-common && \
7
  wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin && \
8
+ mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \
9
+ apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub && \
10
+ add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
11
 
12
  # Install dependencies
13
  RUN apt update && \
 
15
  apt install -y libcublas-dev && \
16
  apt clean && rm -rf /var/lib/apt/lists/*
17
 
 
18
  WORKDIR /app
19
 
20
  # Download ggml and mmproj models from HuggingFace