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
|