AthenaAI / Dockerfile
DucThuanTran's picture
Update Dockerfile
65841d7 verified
# Dockerfile
# Bắt đầu từ image nền Pytorch
FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
# Thiết lập biến môi trường để chỉ định thư mục cache
ENV TRANSFORMERS_CACHE="/app/cache"
ENV HF_HOME="/app/cache"
# Tạo thư mục làm việc
WORKDIR /app
# Tạo trước thư mục cache
RUN mkdir -p /app/cache
# === THÊM DÒNG NÀY ĐỂ GIAO QUYỀN SỞ HỮU THƯ MỤC CACHE ===
# Giao quyền sở hữu thư mục /app/cache cho người dùng sẽ chạy ứng dụng (UID 1000 là user mặc định)
# cờ -R (recursive) đảm bảo áp dụng cho cả các file/thư mục con sẽ được tạo sau này.
RUN chown -R 1000:1000 /app/cache
# =========================================================
# Cài đặt các gói hệ thống
RUN apt-get update && apt-get install -y \
git \
git-lfs \
ffmpeg \
libsndfile1 \
&& rm -rf /var/lib/apt/lists/*
# Sao chép và cài đặt các thư viện Python
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
# Sao chép code ứng dụng
COPY . .
# Mở cổng
EXPOSE 8000
# Lệnh khởi chạy
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]