66 lines
2.2 KiB
Docker
66 lines
2.2 KiB
Docker
# Run command to build:
|
|
# gaudi_ver=$(curl -s "https://vault.habana.ai/artifactory/gaudi-docker/" | sed -n 's/.*href="\([^"]*\).*/\1/p' | tail -2 | head -1 | sed "s/\///1")
|
|
# pytorch_install_ver=$(curl -s "https://vault.habana.ai/artifactory/gaudi-docker/$gaudi_ver/ubuntu20.04/habanalabs/" | sed -n 's/.*href="\([^"]*\).*/\1/p'| sed "s/\///1" | grep pytorch-installer)
|
|
# pytorch_install_ver=${pytorch_install_ver/"pytorch-installer-"/""}
|
|
# docker build -t gaudi-docker-agent:latest \
|
|
# --build-arg GAUDI_VERSION=$gaudi_ver \
|
|
# --build-arg PYTORCH_INSTALLER_VERSION=$pytorch_install_ver \
|
|
# -f Dockerfile .
|
|
# Run command:
|
|
# docker run --privileged \
|
|
# -v /dev:/dev \
|
|
# -e AZP_URL="https://dev.azure.com/ORGANIZATION/" \
|
|
# -e AZP_TOKEN="XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX" \
|
|
# -e AZP_AGENT_NAME="hpu1" \
|
|
# -e AZP_POOL="intel-hpus" \
|
|
# gaudi-docker-agent:latest
|
|
|
|
ARG DIST="latest"
|
|
ARG GAUDI_VERSION="1.7.0"
|
|
ARG PYTORCH_INSTALLER_VERSION="1.12.0"
|
|
FROM vault.habana.ai/gaudi-docker/${GAUDI_VERSION}/ubuntu20.04/habanalabs/pytorch-installer-${PYTORCH_INSTALLER_VERSION}:${DIST}
|
|
|
|
LABEL maintainer="https://vault.habana.ai/"
|
|
# update the base packages and add a non-sudo user
|
|
RUN \
|
|
apt-get update -y && \
|
|
apt-get upgrade -y && \
|
|
useradd -m docker
|
|
|
|
# To make it easier for build and release pipelines to run apt-get,
|
|
# configure apt to not require confirmation (assume the -y argument by default)
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
RUN echo "APT::Get::Assume-Yes \"true\";" > /etc/apt/apt.conf.d/90assumeyes
|
|
|
|
RUN apt-get update --fix-missing && \
|
|
apt-get install -y --no-install-recommends \
|
|
ca-certificates \
|
|
build-essential \
|
|
curl \
|
|
jq \
|
|
git \
|
|
iputils-ping \
|
|
libcurl4 \
|
|
libunwind8 \
|
|
netcat \
|
|
libssl1.0 \
|
|
libssl-dev \
|
|
libffi-dev \
|
|
python3 \
|
|
python3-venv \
|
|
python3-dev \
|
|
python3-pip
|
|
|
|
RUN curl -fsSL https://get.docker.com -o get-docker.sh && \
|
|
sh get-docker.sh && \
|
|
rm get-docker.sh
|
|
|
|
RUN pip uninstall pytorch-lightning -y
|
|
|
|
WORKDIR /azp
|
|
|
|
COPY ./.azure/start.sh /usr/local/bin/
|
|
RUN chmod +x /usr/local/bin/start.sh
|
|
|
|
ENTRYPOINT ["/usr/local/bin/start.sh"]
|