editor / Dockerfile
tanya17's picture
Create Dockerfile
5f35a96 verified
FROM python:3.10
# Install required packages via apt-get
RUN apt-get update && apt-get install -y \
git \
git-lfs \
ffmpeg \
libsm6 \
libxext6 \
cmake \
rsync \
libgl1-mesa-glx \
&& rm -rf /var/lib/apt/lists/* \
&& git lfs install
# Set up a user
RUN apt-get update && apt-get install -y fakeroot && \
mv /usr/bin/apt-get /usr/bin/.apt-get && \
echo '#!/usr/bin/env sh\nfakeroot /usr/bin/.apt-get $@' > /usr/bin/apt-get && \
chmod +x /usr/bin/apt-get && \
rm -rf /var/lib/apt/lists/* && \
useradd -m -u 1000 user
# Set working directory
WORKDIR /home/user/app
# Install pip packages
RUN pip install --no-cache-dir pip==22.3.1 && \
pip install --no-cache-dir \
datasets \
huggingface-hub>=0.19 \
hf-transfer>=0.1.4 \
protobuf<4 \
click<8.1 \
pydantic~=1.0
# Install torch and other Python packages
RUN pip install torch
# Copy requirements.txt and install additional requirements
COPY --chown=1000:1000 requirements.txt /tmp/requirements.txt
RUN pip install --no-cache-dir -r /tmp/requirements.txt
# Set environment variable
ENV OPENAI_API_KEY='/content/openkey.env'