Spaces:
Paused
Paused
Commit
·
98cbe10
1
Parent(s):
44cac5d
Update Dockerfile
Browse files- 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 |
-
|
9 |
-
|
10 |
-
|
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
|