File size: 1,992 Bytes
cbaa28d c2134b3 8e2f1a6 de93d1c c2134b3 8a4c62d cfe22ee 22e21ad cbaa28d bbfea4f fd31e7c e128b40 71e78c9 d94dadb 6cf53aa 7e4f48a 71034a4 b075423 79e7100 c2134b3 e081e60 6cf53aa c2134b3 8e2f1a6 79e7100 6cf53aa c2134b3 4426a67 c2134b3 4426a67 c2134b3 a682a1b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
WORKDIR /code
COPY ./requirements.txt /code/requirements.txt
#RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
#RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht -extra-index-url https://download.pytorch.org/whl/cu111
#https://download.pytorch.org/whl/torch_stable.ht
#https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
RUN apt-get update && apt-get install python3.9 python3-pip git ffmpeg libsm6 libxext6 -y
RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu113
# --index-url https://download.pytorch.org/whl/cu117
# --extra-index-url https://download.pytorch.org/whl/cu113
RUN pip3 install torchvision --extra-index-url https://download.pytorch.org/whl/cu113
RUN pip3 install --no-cache-dir --upgrade -U openmim
RUN mim install --no-cache-dir --upgrade mmengine
RUN mim install "mmcv>=2.0.1"
RUN mim install "mmdet>=3.1.0"
RUN git clone https://github.com/open-mmlab/mmpose.git
# Change directory to mmpose
WORKDIR /code/mmpose
RUN pip3 install -r requirements.txt
RUN pip3 install -e .
# How do we change it back with ..
WORKDIR /..
# RUN mim install "mmpose>=1.1.0"
RUN pip3 install -r /code/requirements.txt
# Set up a new user named "user" with user ID 1000
RUN useradd -m -u 1000 user
# Switch to the "user" user
USER user
# Set home to the user's home directory
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH \
PYTHONPATH=$HOME/app \
PYTHONUNBUFFERED=1 \
GRADIO_ALLOW_FLAGGING=never \
GRADIO_NUM_PORTS=1 \
GRADIO_THEME=huggingface \
SYSTEM=spaces
# Set the working directory to the user's home directory
WORKDIR $HOME/app
# Copy the current directory contents into the container at $HOME/app setting the owner to the user
COPY --chown=user . $HOME/app
CMD ["python3", "main_noweb.py"] |