Spaces:

Manar55
/

truscan

Sleeping

truscan / server /Dockerfile.gpu

Deploy b3083d7

620578c verified 1 day ago

2.12 kB

# TrueScan - GPU server (Hugging Face Spaces GPU, RunPod, any NVIDIA host).
# The PyPI torch wheel for Linux already bundles the CUDA runtime; the host
# only needs an NVIDIA driver (all GPU clouds provide it).
FROM python:3.11-slim

RUN apt-get update && apt-get install -y --no-install-recommends \
        build-essential g++ libgl1 libglib2.0-0 libsm6 libxext6 libxrender1 \
        ffmpeg libsndfile1 git curl \
    && rm -rf /var/lib/apt/lists/*

RUN useradd -m -u 1000 user
ENV HOME=/home/user \
    PIP_NO_CACHE_DIR=1 \
    PIP_DISABLE_PIP_VERSION_CHECK=1 \
    HF_HOME=/home/user/cache/huggingface \
    TORCH_HOME=/home/user/cache/torch \
    XDG_CACHE_HOME=/home/user/cache \
    YOLO_CONFIG_DIR=/home/user/cache/ultralytics \
    MPLCONFIGDIR=/home/user/cache/matplotlib \
    EASYOCR_MODULE_PATH=/home/user/cache/easyocr \
    HF_HUB_DISABLE_SYMLINKS_WARNING=1 \
    PYTHONUNBUFFERED=1 \
    TRUESCAN_SERVER=1 \
    TRUESCAN_PARALLEL=2 \
    PORT=7860

# CUDA 12.8 build: runs on any recent NVIDIA driver (>= 570). Falls back to
# the default PyPI build if that index is unreachable.
RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu128 \
 || pip install torch torchvision
# onnxruntime stays the CPU build on purpose (only insightface uses it; the
# GPU build conflicts with it and depends on a matching CUDA version).
COPY --chown=user server/requirements-server.txt /tmp/req.txt
RUN pip install -r /tmp/req.txt \
 && pip install webrtcvad-wheels \
 && pip install --no-deps resemblyzer \
 && pip install --no-deps "photoholmes @ git+https://github.com/photoholmes/photoholmes" \
 && pip install pydantic pyyaml scikit-image mpmath torchmetrics torch_kmeans typing_extensions \
 && (pip install jpegio || echo "jpegio unavailable") \
 && pip uninstall -y opencv-python \
 && pip install --force-reinstall --no-deps opencv-python-headless

WORKDIR /home/user/app
COPY --chown=user . /home/user/app
ENV PYTHONPATH=/home/user/app:/home/user/app/route:/home/user/app/models
RUN mkdir -p /home/user/cache && chown -R user /home/user

USER user
EXPOSE 7860
CMD ["python", "app.py"]

	# TrueScan - GPU server (Hugging Face Spaces GPU, RunPod, any NVIDIA host).
	# The PyPI torch wheel for Linux already bundles the CUDA runtime; the host
	# only needs an NVIDIA driver (all GPU clouds provide it).
	FROM python:3.11-slim

	RUN apt-get update && apt-get install -y --no-install-recommends \
	build-essential g++ libgl1 libglib2.0-0 libsm6 libxext6 libxrender1 \
	ffmpeg libsndfile1 git curl \
	&& rm -rf /var/lib/apt/lists/*

	RUN useradd -m -u 1000 user
	ENV HOME=/home/user \
	PIP_NO_CACHE_DIR=1 \
	PIP_DISABLE_PIP_VERSION_CHECK=1 \
	HF_HOME=/home/user/cache/huggingface \
	TORCH_HOME=/home/user/cache/torch \
	XDG_CACHE_HOME=/home/user/cache \
	YOLO_CONFIG_DIR=/home/user/cache/ultralytics \
	MPLCONFIGDIR=/home/user/cache/matplotlib \
	EASYOCR_MODULE_PATH=/home/user/cache/easyocr \
	HF_HUB_DISABLE_SYMLINKS_WARNING=1 \
	PYTHONUNBUFFERED=1 \
	TRUESCAN_SERVER=1 \
	TRUESCAN_PARALLEL=2 \
	PORT=7860

	# CUDA 12.8 build: runs on any recent NVIDIA driver (>= 570). Falls back to
	# the default PyPI build if that index is unreachable.
	RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu128 \
	\|\| pip install torch torchvision
	# onnxruntime stays the CPU build on purpose (only insightface uses it; the
	# GPU build conflicts with it and depends on a matching CUDA version).
	COPY --chown=user server/requirements-server.txt /tmp/req.txt
	RUN pip install -r /tmp/req.txt \
	&& pip install webrtcvad-wheels \
	&& pip install --no-deps resemblyzer \
	&& pip install --no-deps "photoholmes @ git+https://github.com/photoholmes/photoholmes" \
	&& pip install pydantic pyyaml scikit-image mpmath torchmetrics torch_kmeans typing_extensions \
	&& (pip install jpegio \|\| echo "jpegio unavailable") \
	&& pip uninstall -y opencv-python \
	&& pip install --force-reinstall --no-deps opencv-python-headless

	WORKDIR /home/user/app
	COPY --chown=user . /home/user/app
	ENV PYTHONPATH=/home/user/app:/home/user/app/route:/home/user/app/models
	RUN mkdir -p /home/user/cache && chown -R user /home/user

	USER user
	EXPOSE 7860
	CMD ["python", "app.py"]