Spaces:
Running
Running
Update Dockerfile
Browse files- Dockerfile +8 -4
Dockerfile
CHANGED
@@ -5,24 +5,28 @@ FROM debian:bullseye-slim
|
|
5 |
RUN dpkg --add-architecture amd64
|
6 |
RUN apt-get update && \
|
7 |
apt-get install -y --no-install-recommends gnupg2 curl ca-certificates
|
8 |
-
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/7fa2af80.pub
|
9 |
|
10 |
# Import the NVIDIA GPG key
|
11 |
#RUN apt-get update && apt-get install -y gnupg2 curl
|
|
|
12 |
RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/7fa2af80.pub | gpg --dearmor -o /usr/share/keyrings/cuda-archive-keyring.gpg
|
13 |
|
14 |
# Add the CUDA repository to sources.list
|
15 |
#RUN echo "deb [arch=amd64 signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" | tee /etc/apt/sources.list.d/cuda.list
|
16 |
-
USER root
|
17 |
RUN echo "deb [arch=amd64 signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" > /etc/apt/sources.list.d/cuda.list
|
18 |
|
|
|
|
|
|
|
19 |
#RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" | sudo tee /etc/apt/sources.list.d/cuda.list
|
20 |
#USER root
|
21 |
#RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" > /etc/apt/sources.list.d/cuda.list
|
22 |
|
23 |
# Install FFmpeg
|
24 |
-
RUN apt-get update && \
|
25 |
-
|
26 |
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
|
27 |
|
28 |
# Use the official Python 3.9 image as the base image
|
|
|
5 |
RUN dpkg --add-architecture amd64
|
6 |
RUN apt-get update && \
|
7 |
apt-get install -y --no-install-recommends gnupg2 curl ca-certificates
|
8 |
+
#RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/7fa2af80.pub
|
9 |
|
10 |
# Import the NVIDIA GPG key
|
11 |
#RUN apt-get update && apt-get install -y gnupg2 curl
|
12 |
+
#RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/7fa2af80.pub | gpg --dearmor -o /usr/share/keyrings/cuda-archive-keyring.gpg
|
13 |
RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/7fa2af80.pub | gpg --dearmor -o /usr/share/keyrings/cuda-archive-keyring.gpg
|
14 |
|
15 |
# Add the CUDA repository to sources.list
|
16 |
#RUN echo "deb [arch=amd64 signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" | tee /etc/apt/sources.list.d/cuda.list
|
17 |
+
#USER root
|
18 |
RUN echo "deb [arch=amd64 signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" > /etc/apt/sources.list.d/cuda.list
|
19 |
|
20 |
+
RUN apt-get update
|
21 |
+
|
22 |
+
|
23 |
#RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" | sudo tee /etc/apt/sources.list.d/cuda.list
|
24 |
#USER root
|
25 |
#RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" > /etc/apt/sources.list.d/cuda.list
|
26 |
|
27 |
# Install FFmpeg
|
28 |
+
#RUN apt-get update && \
|
29 |
+
RUN apt-get install -y python3-pip ffmpeg libcublas-dev
|
30 |
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
|
31 |
|
32 |
# Use the official Python 3.9 image as the base image
|