DucThuanTran commited on
Commit
42f4c96
·
verified ·
1 Parent(s): de77be7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +10 -8
Dockerfile CHANGED
@@ -6,14 +6,19 @@ FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
6
  # Đánh dấu người tạo (tùy chọn)
7
  LABEL maintainer="your-name"
8
 
9
- # === THÊM 2 DÒNG NÀY VÀO ===
10
  # Thiết lập biến môi trường để chỉ định thư mục cache
11
- # Điều này sẽ lưu model vào /app/cache thay vì /.cache, nơi ứng dụng có quyền ghi
12
  ENV TRANSFORMERS_CACHE="/app/cache"
13
  ENV HF_HOME="/app/cache"
14
- # ============================
15
 
16
- # Cài đặt các gói hệ thống cần thiết, ví dụ: git, ffmpeg
 
 
 
 
 
 
 
 
17
  RUN apt-get update && apt-get install -y \
18
  git \
19
  git-lfs \
@@ -21,13 +26,10 @@ RUN apt-get update && apt-get install -y \
21
  libsndfile1 \
22
  && rm -rf /var/lib/apt/lists/*
23
 
24
- # Tạo thư mục làm việc bên trong container
25
- WORKDIR /app
26
-
27
  # Sao chép file requirements.txt vào trước
28
  COPY requirements.txt .
29
 
30
- # Cài đặt các thư viện Python từ file requirements.txt
31
  RUN pip install --no-cache-dir -r requirements.txt
32
 
33
  # Sao chép toàn bộ code của ứng dụng
 
6
  # Đánh dấu người tạo (tùy chọn)
7
  LABEL maintainer="your-name"
8
 
 
9
  # Thiết lập biến môi trường để chỉ định thư mục cache
 
10
  ENV TRANSFORMERS_CACHE="/app/cache"
11
  ENV HF_HOME="/app/cache"
 
12
 
13
+ # Tạo thư mục làm việc bên trong container
14
+ WORKDIR /app
15
+
16
+ # === THÊM DÒNG NÀY VÀO ĐỂ TẠO TRƯỚC THƯ MỤC CACHE ===
17
+ # Lệnh này sẽ tạo thư mục /app/cache với đầy đủ quyền hạn
18
+ RUN mkdir -p /app/cache
19
+ # =========================================================
20
+
21
+ # Cài đặt các gói hệ thống cần thiết
22
  RUN apt-get update && apt-get install -y \
23
  git \
24
  git-lfs \
 
26
  libsndfile1 \
27
  && rm -rf /var/lib/apt/lists/*
28
 
 
 
 
29
  # Sao chép file requirements.txt vào trước
30
  COPY requirements.txt .
31
 
32
+ # Cài đặt các thư viện Python
33
  RUN pip install --no-cache-dir -r requirements.txt
34
 
35
  # Sao chép toàn bộ code của ứng dụng