Spaces:

heboya8
/

test-test

Runtime error

test-test / Dockerfile

Update Dockerfile

6358317 verified over 1 year ago

1.47 kB

# Use NVIDIA CUDA base image for GPU support
FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04

# Set working directory
WORKDIR /app

# Install Python 3.11 and necessary tools
RUN apt-get update --fix-missing && apt-get install -y --no-install-recommends \
    python3.11 \
    python3.11-dev \
    python3-pip \
    git \
    gcc \
    build-essential \
    ffmpeg \
    && rm -rf /var/lib/apt/lists/* \
    && ln -sf /usr/bin/python3.11 /usr/bin/python3 \
    && ln -sf /usr/bin/python3.11 /usr/bin/python

# Set Hugging Face cache directory
ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
ENV HF_HOME=/app/.cache/huggingface
RUN mkdir -p /app/.cache/huggingface && chmod -R 777 /app/.cache/huggingface

# Copy requirements file
COPY requirements.txt .

# Update pip and install pip-tools
RUN pip install --no-cache-dir --upgrade pip \
    && pip install --no-cache-dir pip-tools

# Sync dependencies with pip-sync
RUN pip-sync requirements.txt

# Install Gradio, diffusers, and LoRA packages
RUN pip install --no-cache-dir gradio==4.44.0
RUN pip install --no-cache-dir diffusers==0.30.3
RUN pip install --no-cache-dir huggingface_hub==0.25.2
RUN pip install git+https://github.com/cloneofsimo/lora.git
RUN pip install git+https://github.com/microsoft/LoRA

# Copy the rest of the application code
COPY . .

# Expose Gradio default port
EXPOSE 7860

# Set environment variable to prevent Python buffering
ENV PYTHONUNBUFFERED=1

# Run the Gradio app
CMD ["python", "app.py"]

	# Use NVIDIA CUDA base image for GPU support
	FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04

	# Set working directory
	WORKDIR /app

	# Install Python 3.11 and necessary tools
	RUN apt-get update --fix-missing && apt-get install -y --no-install-recommends \
	python3.11 \
	python3.11-dev \
	python3-pip \
	git \
	gcc \
	build-essential \
	ffmpeg \
	&& rm -rf /var/lib/apt/lists/* \
	&& ln -sf /usr/bin/python3.11 /usr/bin/python3 \
	&& ln -sf /usr/bin/python3.11 /usr/bin/python

	# Set Hugging Face cache directory
	ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
	ENV HF_HOME=/app/.cache/huggingface
	RUN mkdir -p /app/.cache/huggingface && chmod -R 777 /app/.cache/huggingface

	# Copy requirements file
	COPY requirements.txt .

	# Update pip and install pip-tools
	RUN pip install --no-cache-dir --upgrade pip \
	&& pip install --no-cache-dir pip-tools

	# Sync dependencies with pip-sync
	RUN pip-sync requirements.txt

	# Install Gradio, diffusers, and LoRA packages
	RUN pip install --no-cache-dir gradio==4.44.0
	RUN pip install --no-cache-dir diffusers==0.30.3
	RUN pip install --no-cache-dir huggingface_hub==0.25.2
	RUN pip install git+https://github.com/cloneofsimo/lora.git
	RUN pip install git+https://github.com/microsoft/LoRA

	# Copy the rest of the application code
	COPY . .

	# Expose Gradio default port
	EXPOSE 7860

	# Set environment variable to prevent Python buffering
	ENV PYTHONUNBUFFERED=1

	# Run the Gradio app
	CMD ["python", "app.py"]