diff --git a/Dockerfile.dev b/Dockerfile.dev new file mode 100644 index 0000000..4a477f6 --- /dev/null +++ b/Dockerfile.dev @@ -0,0 +1,33 @@ +# 使用官方的paddle镜像作为基础 +FROM ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlex/paddlex:paddlex3.1.2-paddlepaddle3.0.0-gpu-cuda12.6-cudnn9.5-trt10.5 + +# 设置工作目录 +WORKDIR /app + +# 设置环境变量 +ENV PYTHONUNBUFFERED=1 \ + # 设置时区 + TZ=Asia/Shanghai \ + # 设置pip镜像地址,加快安装速度 + PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple + +# 安装language-pack-en和openssh-server +RUN apt update && \ + apt install -y language-pack-en && \ + apt install -y openssh-server + +# 配置SSH服务 +RUN mkdir /var/run/sshd && \ + # 设置root密码,可根据需要修改 + echo 'root:fcb0102' | chpasswd && \ + # 允许root登录SSH + sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config + +# 将当前目录内容复制到容器的/app内 +COPY . /app + +# 暴露22端口 +EXPOSE 22 + +# 启动SSH服务 +CMD ["/usr/sbin/sshd", "-D"] \ No newline at end of file diff --git a/docker-compose.dev.yml b/docker-compose.dev.yml new file mode 100644 index 0000000..a21c6f0 --- /dev/null +++ b/docker-compose.dev.yml @@ -0,0 +1,26 @@ +services: + fcb_ai_dev: + image: fcb_ai_dev:0.0.7 + build: + context: . + dockerfile: Dockerfile.dev + # 容器名称,可自定义 + container_name: fcb_ai_dev + hostname: fcb_ai_dev + # 始终重启容器 + restart: always + # 端口映射,根据需要修改主机端口 + ports: + - "8022:22" + # 数据卷映射,根据实际路径修改 + volumes: + - ./log:/app/log + - ./model:/app/model + # 启用GPU支持 + deploy: + resources: + reservations: + devices: + - device_ids: [ '0', '1' ] + capabilities: [ 'gpu' ] + driver: 'nvidia' \ No newline at end of file