fffiloni commited on
Commit
164e345
·
verified ·
1 Parent(s): 2630fab

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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-7 libcudnn8 libcudnn8-dev
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 --no-binary dlib
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 .