VioMobileNet / Dockerfile
quochuy
update gpu
54ea07d
raw
history blame contribute delete
444 Bytes
FROM tensorflow/tensorflow:2.15.0-gpu
RUN apt-get update && apt-get install -y libgl1-mesa-glx libglib2.0-0 && rm -rf /var/lib/apt/lists/*
WORKDIR /app
# Copy code
COPY . .
# Install dependencies
RUN pip install "numpy<2" opencv-python fastapi uvicorn[standard] websockets
ENV MODEL_PATH="model"
ENV RESOLUTION="172"
# Mở port 8000
EXPOSE 8000
# Chạy Server FastAPI
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]