Commit
•
8e550d8
1
Parent(s):
4c8517b
Update Dockerfile
Browse files- Dockerfile +4 -6
Dockerfile
CHANGED
|
@@ -4,17 +4,14 @@ ARG DEBIAN_FRONTEND=noninteractive
|
|
| 4 |
|
| 5 |
ENV PYTHONUNBUFFERED=1
|
| 6 |
|
| 7 |
-
RUN apt-get update && apt-get install
|
| 8 |
build-essential \
|
| 9 |
python3.9 \
|
| 10 |
python3-pip \
|
| 11 |
git \
|
| 12 |
-
ffmpeg
|
| 13 |
-
|
| 14 |
-
RUN apt-get update && apt-get install --no-install-recommends -y \
|
| 15 |
libcudnn8 \
|
| 16 |
-
libcudnn8-dev
|
| 17 |
-
|
| 18 |
|
| 19 |
WORKDIR /code
|
| 20 |
|
|
@@ -39,6 +36,7 @@ ENV HOME=/home/user \
|
|
| 39 |
GRADIO_THEME=huggingface \
|
| 40 |
SYSTEM=spaces
|
| 41 |
|
|
|
|
| 42 |
RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 43 |
|
| 44 |
# Set the working directory to the user's home directory
|
|
|
|
| 4 |
|
| 5 |
ENV PYTHONUNBUFFERED=1
|
| 6 |
|
| 7 |
+
RUN apt-get update && apt-get install -y \
|
| 8 |
build-essential \
|
| 9 |
python3.9 \
|
| 10 |
python3-pip \
|
| 11 |
git \
|
| 12 |
+
ffmpeg \
|
|
|
|
|
|
|
| 13 |
libcudnn8 \
|
| 14 |
+
libcudnn8-dev
|
|
|
|
| 15 |
|
| 16 |
WORKDIR /code
|
| 17 |
|
|
|
|
| 36 |
GRADIO_THEME=huggingface \
|
| 37 |
SYSTEM=spaces
|
| 38 |
|
| 39 |
+
RUN export LD_LIBRARY_PATH=`python3 -c 'import os; import nvidia.cublas.lib; import nvidia.cudnn.lib; import torch; print(os.path.dirname(nvidia.cublas.lib.__file__) + ":" + os.path.dirname(nvidia.cudnn.lib.__file__) + ":" + os.path.dirname(torch.__file__) +"/lib")'`:$LD_LIBRARY_PATH
|
| 40 |
RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 41 |
|
| 42 |
# Set the working directory to the user's home directory
|