mirror of
https://github.com/openai/whisper.git
synced 2025-11-24 06:26:03 +00:00
38 lines
975 B
Docker
38 lines
975 B
Docker
# Use the official Gaudi Docker image with PyTorch
|
|
FROM vault.habana.ai/gaudi-docker/1.17.0/ubuntu22.04/habanalabs/pytorch-installer-2.3.1: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
|
|
|
|
# Install essential Linux packages
|
|
ENV DEBIAN_FRONTEND="noninteractive" TZ=Etc/UTC
|
|
RUN apt-get update && apt-get install -y \
|
|
tzdata \
|
|
bash-completion \
|
|
python3-pip \
|
|
openssh-server \
|
|
vim \
|
|
iputils-ping \
|
|
net-tools \
|
|
protobuf-compiler \
|
|
curl \
|
|
bc \
|
|
gawk \
|
|
tmux \
|
|
ffmpeg \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Add Whisper repo contents
|
|
ADD . /root/whisper
|
|
WORKDIR /root/whisper
|
|
|
|
# Copy requirements.txt explicitly
|
|
COPY requirements.txt /root/whisper/requirements.txt
|
|
|
|
# Install Python packages from requirements.txt
|
|
RUN pip install --upgrade pip \
|
|
&& pip install -r requirements.txt
|