18 lines
505 B
Docker
18 lines
505 B
Docker
|
# 使用基础镜像
|
|||
|
|
FROM pytorch/pytorch:2.4.1-cuda11.8-cudnn9-runtime
|
|||
|
|
|
|||
|
|
# 设置工作目录
|
|||
|
|
WORKDIR /app
|
|||
|
|
|
|||
|
|
# 复制 requirements.txt 文件（这一步如果文件没变，Docker 会使用缓存）
|
|||
|
|
COPY requirements.txt /app/requirements.txt
|
|||
|
|
|
|||
|
|
# 安装依赖（Docker 会缓存这一步，除非 requirements.txt 发生变化）
|
|||
|
|
RUN pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
|
|||
|
|
|
|||
|
|
# 复制代码到容器中
|
|||
|
|
COPY ./src /app/src
|
|||
|
|
|
|||
|
|
# 运行应用
|
|||
|
|
CMD ["python", "-m", "src.api"]
|