26 lines
804 B
Docker
26 lines
804 B
Docker
# Use the official Gaudi Docker image with PyTorch
|
|
FROM vault.habana.ai/gaudi-docker/1.18.0/ubuntu22.04/habanalabs/pytorch-installer-2.4.0:latest
|
|
|
|
# Set environment variables for Habana
|
|
ENV HABANA_VISIBLE_DEVICES=all
|
|
ENV OMPI_MCA_btl_vader_single_copy_mechanism=none
|
|
ENV PT_HPU_LAZY_ACC_PAR_MODE=0
|
|
ENV PT_HPU_ENABLE_LAZY_COLLECTIVES=1
|
|
|
|
# Set timezone to UTC and install essential packages
|
|
ENV DEBIAN_FRONTEND="noninteractive" TZ=Etc/UTC
|
|
RUN apt-get update && apt-get install -y \
|
|
tzdata \
|
|
python3-pip \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
COPY . /workspace/clip
|
|
WORKDIR /workspace/clip
|
|
|
|
# Copy HPU requirements
|
|
COPY requirements_hpu.txt /workspace/requirements_hpu.txt
|
|
|
|
# Install Python packages
|
|
RUN pip install --upgrade pip \
|
|
&& pip install -r requirements_hpu.txt \
|
|
&& pip install -e . |