search

deep-dance / core

GNU General Public License v3.0
2 stars 1 forks source link

Create docker container supporting all modules #12

Closed erak closed 3 years ago

erak commented 3 years ago 
FROM nvidia/cuda:10.1-base

CMD nvidia-smi

RUN apt-get update
RUN apt-get update
RUN apt-get install -y python3-venv python3-pip git

RUN python3 -m pip install detectron2 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.7/index.html

RUN pip3 install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
RUN pip3 install opencv-python

RUN git clone --recursive https://github.com/zirkular/to-share.git
RUN cd to-share

RUN cd detector/VideoPose3D
RUN mkdir checkpoint && cd checkpoint
RUN wget https://dl.fbaipublicfiles.com/video-pose-3d/pretrained_h36m_detectron_coco.bin

RUN cd ..
RUN cd ..
RUN cd ..
nikozoe commented 3 years ago

added Dockerfile and documentation.