2024-09-11 01:09:35 +08:00
|
|
|
|
# 使用基础镜像
|
|
|
|
|
|
FROM pytorch/pytorch:2.4.1-cuda11.8-cudnn9-runtime
|
|
|
|
|
|
|
|
|
|
|
|
# 设置工作目录
|
|
|
|
|
|
WORKDIR /app
|
|
|
|
|
|
|
|
|
|
|
|
# 复制 requirements.txt 文件（这一步如果文件没变，Docker 会使用缓存）
|
2024-10-08 22:16:17 +08:00
|
|
|
|
COPY ../requirements.txt /app/requirements.txt
|
2024-09-11 01:09:35 +08:00
|
|
|
|
|
|
|
|
|
|
# 安装依赖（Docker 会缓存这一步，除非 requirements.txt 发生变化）
|
|
|
|
|
|
RUN pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
|
|
|
|
|
|
|
|
|
|
|
|
# 复制代码到容器中
|
2024-10-08 22:16:17 +08:00
|
|
|
|
COPY ../src /app/src
|
2024-09-11 01:09:35 +08:00
|
|
|
|
|
