car-detection-bayes/Dockerfile

63 lines
2.6 KiB
Docker
Raw Normal View History

2019-11-30 20:43:41 +00:00
# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
2020-01-29 18:29:37 +00:00
FROM nvcr.io/nvidia/pytorch:20.01-py3
2019-08-29 16:58:09 +00:00
2019-08-29 18:13:29 +00:00
# Install dependencies (pip or conda)
2019-09-10 10:40:59 +00:00
RUN pip install -U gsutil
2019-08-29 17:57:08 +00:00
# RUN pip install -U -r requirements.txt
2019-08-29 18:13:29 +00:00
# RUN conda update -n base -c defaults conda
# RUN conda install -y -c anaconda future numpy opencv matplotlib tqdm pillow
# RUN conda install -y -c conda-forge scikit-image tensorboard pycocotools
2019-08-29 16:58:09 +00:00
2019-09-16 12:31:07 +00:00
## Install OpenCV with Gstreamer support
#WORKDIR /usr/src
#RUN pip uninstall -y opencv-python
#RUN apt-get update
2019-09-16 12:31:07 +00:00
#RUN apt-get install -y gstreamer1.0-tools gstreamer1.0-python3-dbg-plugin-loader libgstreamer1.0-dev libgstreamer-plugins-base1.0-dev
#RUN git clone https://github.com/opencv/opencv.git && cd opencv && git checkout 4.1.1 && mkdir build
#RUN git clone https://github.com/opencv/opencv_contrib.git && cd opencv_contrib && git checkout 4.1.1
#RUN cd opencv/build && cmake ../ \
# -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules \
# -D BUILD_OPENCV_PYTHON3=ON \
# -D PYTHON3_EXECUTABLE=/opt/conda/bin/python \
# -D PYTHON3_INCLUDE_PATH=/opt/conda/include/python3.6m \
# -D PYTHON3_LIBRARIES=/opt/conda/lib/python3.6/site-packages \
# -D WITH_GSTREAMER=ON \
2019-09-16 12:31:07 +00:00
# -D WITH_FFMPEG=OFF \
# && make && make install && ldconfig
#RUN cd /usr/local/lib/python3.6/site-packages/cv2/python-3.6/ && mv cv2.cpython-36m-x86_64-linux-gnu.so cv2.so
#RUN cd /opt/conda/lib/python3.6/site-packages/ && ln -s /usr/local/lib/python3.6/site-packages/cv2/python-3.6/cv2.so cv2.so
#RUN python3 -c "import cv2; print(cv2.getBuildInformation())"
# Create working directory
RUN mkdir -p /usr/src/app
WORKDIR /usr/src/app
2019-08-29 16:58:09 +00:00
# Copy contents
COPY . /usr/src/app
2019-08-29 16:58:09 +00:00
2019-09-09 17:15:17 +00:00
# Copy weights
2019-12-07 01:31:07 +00:00
#RUN python3 -c "from models import *; \
#attempt_download('weights/yolov3.pt'); \
#attempt_download('weights/yolov3-spp.pt')"
2019-09-09 17:15:17 +00:00
2019-08-29 16:58:09 +00:00
# --------------------------------------------------- Extras Below ---------------------------------------------------
2019-12-07 23:06:38 +00:00
# Build and Push
2019-12-22 03:39:45 +00:00
# t=ultralytics/yolov3:v0 && sudo docker build -t $t . && sudo docker push $t
2019-08-29 16:58:09 +00:00
2019-09-16 19:08:14 +00:00
# Run
2020-03-08 18:56:37 +00:00
# t=ultralytics/yolov3:v0 && sudo docker pull $t && sudo docker run -it --gpus all --ipc=host $t bash
2019-10-09 01:16:27 +00:00
# Pull and Run with local directory access
2020-03-08 18:56:37 +00:00
# t=ultralytics/yolov3:v0 && sudo docker pull $t && sudo docker run -it --gpus all --ipc=host -v "$(pwd)"/coco:/usr/src/coco $t bash
2019-10-09 01:16:27 +00:00
2019-09-16 19:08:14 +00:00
# Kill all
2019-11-30 21:20:22 +00:00
# sudo docker kill "$(sudo docker ps -q)"
2019-09-09 17:15:17 +00:00
2019-12-27 18:08:58 +00:00
# Kill all image-based
# sudo docker kill $(sudo docker ps -a -q --filter ancestor=ultralytics/yolov3:v0)
2019-09-10 15:04:33 +00:00
# Run bash for loop
2020-01-15 19:22:06 +00:00
# sudo docker run --gpus all --ipc=host ultralytics/yolov3:v0 while true; do python3 train.py --evolve; done