chore: change docker env to ubuntu 20.04

This commit is contained in:
wql 2024-08-29 11:19:48 +08:00
parent e0a4a0e451
commit d2a4d909c1
1 changed files with 4 additions and 4 deletions

View File

@ -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 # 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-910-ubuntu20.04-py3.8
FROM ascendai/cann:8.0.rc1-910b-ubuntu22.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-910-openeuler22.03-py3.8
# FROM ascendai/cann:8.0.rc1-910b-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 # Define installation arguments
ARG INSTALL_DEEPSPEED=false 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 ARG TORCH_INDEX=https://download.pytorch.org/whl/cpu
# Set the working directory # Set the working directory