From d2a4d909c163a1fbc23493aed8d49a07a3a28151 Mon Sep 17 00:00:00 2001 From: wql Date: Thu, 29 Aug 2024 11:19:48 +0800 Subject: [PATCH] chore: change docker env to ubuntu 20.04 --- docker/docker-npu/Dockerfile | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/docker/docker-npu/Dockerfile b/docker/docker-npu/Dockerfile index dc35de47..3075a023 100644 --- a/docker/docker-npu/Dockerfile +++ b/docker/docker-npu/Dockerfile @@ -1,7 +1,7 @@ -# Use the Ubuntu 22.04 image with CANN 8.0.rc1 +# Use the Ubuntu 20.04 image with CANN 8.0.rc1 # More versions can be found at https://hub.docker.com/r/ascendai/cann/tags -# FROM ascendai/cann:8.0.rc1-910-ubuntu22.04-py3.8 -FROM ascendai/cann:8.0.rc1-910b-ubuntu22.04-py3.8 +# FROM ascendai/cann:8.0.rc1-910-ubuntu20.04-py3.8 +FROM ascendai/cann:8.0.rc1-910b-ubuntu20.04-py3.8 # FROM ascendai/cann:8.0.rc1-910-openeuler22.03-py3.8 # FROM ascendai/cann:8.0.rc1-910b-openeuler22.03-py3.8 @@ -10,7 +10,7 @@ ENV DEBIAN_FRONTEND=noninteractive # Define installation arguments ARG INSTALL_DEEPSPEED=false -ARG PIP_INDEX=https://pypi.org/simple +ARG PIP_INDEX=https://pypi.tuna.tsinghua.edu.cn/simple ARG TORCH_INDEX=https://download.pytorch.org/whl/cpu # Set the working directory