first docker

docker/Dockerfile

@@ -1,7 +1,31 @@
-FROM ceccocats/tkdnn:latest
-LABEL maintainer "Francesco Gatti"
+FROM mohitkhubele95/tkdnn
+ARG DEBIAN_FRONTEND=noninteractive
 
-RUN cd && git clone https://github.com/ceccocats/tkDNN.git && cd tkDNN && mkdir build && cd build \ 
-    && cmake .. && make -j12
+RUN apt-get update
+RUN apt-get install -y software-properties-common
+RUN add-apt-repository 'deb http://security.ubuntu.com/ubuntu xenial-security main'
+RUN apt-get -y update
+RUN apt-get -y upgrade
 
+RUN apt-get -y install cmake g++ git sudo vim curl rapidjson-dev awscli zip unzip dpkg libcpprest-dev libboost-dev libboost-all-dev
 
+RUN  useradd -ms /bin/bash baggageai && echo "baggageai:baggageai" | chpasswd && adduser baggageai sudo
+USER baggageai
+WORKDIR /home/baggageai
+EXPOSE 8080
+
+#RUN aws s3 cp s3://dim-bai-s3-dev-developer-space/smiths_29_objects/BaggageAI.zip .
+
+RUN mkdir files
+#RUN mkdir files/include
+#RUN mkdir files/server
+
+#Change path of include and server folder accordingly
+#COPY --chown=baggageai:baggageai src/include/ files/include
+#COPY --chown=baggageai:baggageai src/server/ files/server
+COPY --chown=baggageai:baggageai . files/
+#RUN aws s3 cp s3://dim-bai-s3-dev-developer-space/smiths_29_objects/libBaggageAI.so files/
+
+WORKDIR /home/baggageai/files
+#RUN chmod 777 run.sh
+#ENTRYPOINT ["./run.sh"]

docker/Dockerfile.base

@@ -1,57 +0,0 @@
-FROM nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04
-LABEL maintainer "Francesco Gatti"
-
-ADD nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.0.0.11-ga-20191216_1-1_amd64.deb /tmp/trt.deb
-RUN apt-get update && dpkg -i /tmp/trt.deb && rm /tmp/trt.deb && apt-get update
-RUN apt install -y libnvinfer7=7.0.0-1+cuda10.2 libnvinfer-dev=7.0.0-1+cuda10.2
-RUN DEBIAN_FRONTEND=noninteractive apt install -y git wget libeigen3-dev libyaml-cpp-dev
-RUN cd /tmp && \
-    wget https://github.com/Kitware/CMake/releases/download/v3.17.3/cmake-3.17.3-Linux-x86_64.sh && \
-    chmod +x cmake-3.17.3-Linux-x86_64.sh && \
-    ./cmake-3.17.3-Linux-x86_64.sh --prefix=/usr/local --exclude-subdir --skip-license && \
-    rm ./cmake-3.17.3-Linux-x86_64.sh
-
-RUN echo "INSTALL OPENCV"
-RUN apt-get install -y build-essential \
-    unzip \
-    pkg-config \
-    libjpeg-dev \
-    libpng-dev \
-    libtiff-dev \
-    libavcodec-dev \
-    libavformat-dev \
-    libswscale-dev \
-    libv4l-dev \
-    libxvidcore-dev \
-    libx264-dev \
-    libgtk-3-dev \
-    libatlas-base-dev \
-    gfortran \
-    libgstreamer1.0-dev \
-    libgstreamer-plugins-base1.0-dev \
-    libdc1394-22-dev \
-    libavresample-dev
-RUN cd && wget https://github.com/opencv/opencv/archive/4.3.0.tar.gz && tar -xf 4.3.0.tar.gz && rm *.tar.gz
-RUN cd && wget https://github.com/opencv/opencv_contrib/archive/4.3.0.tar.gz && tar -xf 4.3.0.tar.gz && rm *.tar.gz
-RUN cd && \
-    cd opencv-4.3.0 && mkdir build && cd build && \
-    cmake -D CMAKE_BUILD_TYPE=RELEASE \
-    -D CMAKE_INSTALL_PREFIX=/usr/local \
-    -D INSTALL_PYTHON_EXAMPLES=OFF \
-    -D INSTALL_C_EXAMPLES=OFF \
-    -D OPENCV_EXTRA_MODULES_PATH='~/opencv_contrib-4.3.0/modules' \
-    -D BUILD_EXAMPLES=OFF \
-    -D WITH_CUDA=ON \
-    -D CUDA_ARCH_BIN=7.2 \
-    -D CUDA_ARCH_PTX="" \
-    -D ENABLE_FAST_MATH=ON \
-    -D CUDA_FAST_MATH=ON \
-    -D WITH_CUBLAS=ON \
-    -D WITH_LIBV4L=ON \
-    -D WITH_GSTREAMER=ON \
-    -D WITH_GSTREAMER_0_10=OFF \
-    -D WITH_TBB=ON \
-    ../ && make -j12 && make install
-RUN apt clean
-
-

docker/README.md

@@ -1,21 +0,0 @@
-# Use the prebuilt image
-```
-# build image
-docker build -t tkdnn:build -f Dockerfile . 
-```
-
-# Build Base Docker image
-```
-# make nvidia docker working
-# follow this guide: https://github.com/NVIDIA/nvidia-docker
-
-# dowload tensorrt
-# from: https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/7.0/7.0.0.11/local_repo/nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.0.0.11-ga-20191216_1-1_amd64.deb
-
-# build image
-docker build -t ceccocats/tkdnn:latest -f Dockerfile.base .
-
-# run image
-docker run -ti --gpus all --rm ceccocats/tkdnn:latest bash
-```
-