Update Dockerfile

6295954b · dcuai · 2d952503 · 6295954b
Commit 6295954b authored Sep 19, 2024 by dcuai
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

docker/Dockerfile docker/Dockerfile +2 -2

No files found.
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
-FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py38-latest
+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
 ENV DEBIAN_FRONTEND=noninteractive
 # RUN yum update && yum install -y git cmake wget build-essential
-RUN source /opt/dtk-23.04/env.sh
+RUN source /opt/dtk/env.sh
 # 安装pip相关依赖
 COPY requirements.txt requirements.txt
 RUN pip3 install -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com -r requirements.txt