Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +12 -2
Dockerfile
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
FROM
|
| 2 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 3 |
|
| 4 |
RUN apt-get update && apt-get install -y \
|
|
@@ -27,7 +27,17 @@ USER user
|
|
| 27 |
ENV HOME=/home/user \
|
| 28 |
PATH=/home/user/.local/bin:$PATH \
|
| 29 |
PYTHONPATH=$HOME/app \
|
| 30 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
|
| 32 |
# Clone the RB-Modulation repository
|
| 33 |
RUN git clone https://github.com/google/RB-Modulation.git $HOME/app
|
|
|
|
| 1 |
+
FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-devel
|
| 2 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 3 |
|
| 4 |
RUN apt-get update && apt-get install -y \
|
|
|
|
| 27 |
ENV HOME=/home/user \
|
| 28 |
PATH=/home/user/.local/bin:$PATH \
|
| 29 |
PYTHONPATH=$HOME/app \
|
| 30 |
+
PYTHONUNBUFFERED=1 \
|
| 31 |
+
GRADIO_ALLOW_FLAGGING=never \
|
| 32 |
+
GRADIO_NUM_PORTS=1 \
|
| 33 |
+
GRADIO_SERVER_NAME=0.0.0.0 \
|
| 34 |
+
GRADIO_THEME=huggingface \
|
| 35 |
+
GRADIO_SHARE=False \
|
| 36 |
+
SYSTEM=spaces
|
| 37 |
+
|
| 38 |
+
# Set the environment variable to specify the GPU device
|
| 39 |
+
ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
|
| 40 |
+
ENV CUDA_VISIBLE_DEVICES=0
|
| 41 |
|
| 42 |
# Clone the RB-Modulation repository
|
| 43 |
RUN git clone https://github.com/google/RB-Modulation.git $HOME/app
|