YOLO-World4 / Dockerfile
csmithxc's picture
Update Dockerfile
1897a05 verified
FROM nvidia/cuda:12.4.1-devel-ubuntu22.04
RUN curl https://pyenv.run | bash
COPY --chown=1000:1000 --from=root / /
RUN pyenv install 3.10 && pyenv global 3.10 && pyenv rehash
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
WORKDIR /home/user/app
RUN apt-get update && apt-get install -y git rsync make build-essential libssl-dev zlib1g-dev \
libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev git-lfs \
ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
&& rm -rf /var/lib/apt/lists/* \
&& git lfs install
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" "torch>=2.1.0"
RUN --mount=target=/tmp/requirements.txt,source=requirements.txt \
pip install --no-cache-dir -r /tmp/requirements.txt