| 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 | |
| # 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"] |