fiewolf1000 commited on
Commit
ae7a9bf
·
verified ·
1 Parent(s): b010be7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +10 -11
Dockerfile CHANGED
@@ -1,32 +1,31 @@
1
- # 基础镜像(Python 3.10 稳定版)
2
  FROM python:3.10-slim
3
 
4
- # 设置工作目录
5
  WORKDIR /app
6
 
7
- # 安装系统依赖(编译 PyTorch 所需)
8
  RUN apt-get update && apt-get install -y --no-install-recommends \
9
  gcc \
10
  g++ \
11
  && rm -rf /var/lib/apt/lists/*
12
 
13
- # 升级 pip(解决依赖解析问题)
14
  RUN pip install --no-cache-dir --upgrade pip
15
 
16
- # 创建可写的 Hugging Face 缓存目录(解决权限问题)
17
- RUN mkdir -p /tmp/huggingface_cache && chmod 777 /tmp/huggingface_cache
18
  ENV TRANSFORMERS_CACHE=/tmp/huggingface_cache \
19
- HUGGINGFACE_HUB_CACHE=/tmp/huggingface_cache
 
 
20
 
21
- # 安装 Python 依赖
22
  COPY requirements.txt .
23
  RUN pip install --no-cache-dir -r requirements.txt
24
 
25
- # 复制应用代码
26
  COPY . .
27
 
28
- # 暴露端口(与 app.py 中一致)
29
  EXPOSE 7860
30
 
31
- # 启动服务
32
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
 
1
+ # 基础镜像
2
  FROM python:3.10-slim
3
 
 
4
  WORKDIR /app
5
 
6
+ # 安装系统依赖
7
  RUN apt-get update && apt-get install -y --no-install-recommends \
8
  gcc \
9
  g++ \
10
  && rm -rf /var/lib/apt/lists/*
11
 
12
+ # 升级 pip
13
  RUN pip install --no-cache-dir --upgrade pip
14
 
15
+ # 【关键修改】创建缓存目录并设置 777 权限(所有用户可读写)
16
+ RUN mkdir -p /tmp/huggingface_cache && chmod -R 777 /tmp/huggingface_cache
17
  ENV TRANSFORMERS_CACHE=/tmp/huggingface_cache \
18
+ HUGGINGFACE_HUB_CACHE=/tmp/huggingface_cache \
19
+ # 额外添加缓存路径变量,确保所有库都使用该目录
20
+ HF_HOME=/tmp/huggingface_cache
21
 
22
+ # 安装依赖
23
  COPY requirements.txt .
24
  RUN pip install --no-cache-dir -r requirements.txt
25
 
26
+ # 复制代码
27
  COPY . .
28
 
 
29
  EXPOSE 7860
30
 
 
31
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]