File size: 417 Bytes
29d411b |
1 2 3 4 5 6 7 8 9 10 11 12 |
FROM nvcr.io/nvidia/pytorch:24.07-py3
COPY . /easy_ViTPose
WORKDIR /easy_ViTPose/
ENV DEBIAN_FRONTEND=noninteractive
RUN pip uninstall -y $(pip list --format=freeze | grep opencv) && \
rm -rf /usr/local/lib/python3.10/dist-packages/cv2/
RUN pip install -e . && pip install -r requirements.txt && pip install -r requirements_gpu.txt
# OpenCV dependency
RUN apt-get update && apt-get install -y libgl1
|