Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
@@ -16,7 +16,7 @@ RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86
|
|
16 |
apt-get update
|
17 |
|
18 |
# Install CUDA Toolkit and cuDNN
|
19 |
-
RUN apt-get install -y cuda-toolkit-11-
|
20 |
|
21 |
# Create a user
|
22 |
RUN useradd -m -u 1000 user
|
@@ -59,7 +59,7 @@ RUN pip install --no-cache-dir urllib3 transformers yacs scipy scikit-image scik
|
|
59 |
RUN pip install cmake
|
60 |
|
61 |
# Install dlib with CUDA support (force source compilation)
|
62 |
-
RUN pip install --no-cache-dir dlib -vvv
|
63 |
|
64 |
# Copy application file
|
65 |
COPY app.py .
|
|
|
16 |
apt-get update
|
17 |
|
18 |
# Install CUDA Toolkit and cuDNN
|
19 |
+
RUN apt-get install -y cuda-toolkit-11-8 libcudnn8 libcudnn8-dev
|
20 |
|
21 |
# Create a user
|
22 |
RUN useradd -m -u 1000 user
|
|
|
59 |
RUN pip install cmake
|
60 |
|
61 |
# Install dlib with CUDA support (force source compilation)
|
62 |
+
RUN pip install --no-cache-dir dlib -vvv
|
63 |
|
64 |
# Copy application file
|
65 |
COPY app.py .
|