diff --git a/Dockerfile.mlu370 b/Dockerfile.mlu370 index d8a5a55..9131487 100644 --- a/Dockerfile.mlu370 +++ b/Dockerfile.mlu370 @@ -1,4 +1,4 @@ -FROM cambricon-base/pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310 +FROM git.modelhub.org.cn:9443/enginex-cambricon/mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310 WORKDIR /workspace ENV PATH=/torch/venv3/pytorch_infer/bin:/workspace/ffmpeg-mlu-v4.2.0/install/bin:/usr/local/neuware/bin:/usr/local/openmpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin