Commit 9bc751e7 authored by fierval's avatar fierval
Browse files

docker stuff

parent 4dfdcc86
Loading
Loading
Loading
Loading

cpall.sh

100644 → 100755
+0 −0

File mode changed from 100644 to 100755.

docker/Dockerfile

0 → 100644
+89 −0
Original line number Diff line number Diff line
FROM fierval/tensorrt:19.02-py3

# tensorflow libraries
ADD tensorflow.tar /

# nvcuvid
ADD nvcuvid/* /usr/local/cuda/targets/x86_64-linux/include/

# opencv
RUN apt-get update
RUN apt-get install -y git libgtk2.0-dev curl pkg-config libavcodec-dev \
libavformat-dev libswscale-dev python-dev python-numpy libtbb2 libtbb-dev \
libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev unzip libcurl4-gnutls-dev zlib1g-dev

RUN apt-get install -y wget

RUN apt-get install -y vim

## CMAKE
ADD https://cmake.org/files/v3.13/cmake-3.13.0.tar.gz /
RUN tar xzvf /cmake-3.13.0.tar.gz -C / \
    && cd /cmake-3.13.0 \
    && ./bootstrap \
    && make -j15 \
    && make install

# Second: get and build OpenCV 4.0.1
#

ADD https://github.com/opencv/opencv_contrib/archive/4.0.1.zip  / 
RUN unzip -o /4.0.1.zip
#     && cd opencv-4.0.1/build \
#     && cmake -DOPENCV_EXTRA_MODULES_PATH=../../opencv_contrib-4.0.1/modules/ .. \
#     && make -j8 \
#     && make install \
#     && cd ../.. \
#     && rm 4.0.1.zip

ADD https://github.com/opencv/opencv/archive/4.0.1.zip / 
RUN unzip -o /4.0.1.zip 
RUN cd /workspace/opencv-4.0.1 \
    && mkdir build \
    && cd build \
    && cmake -DBUILD_TIFF=ON \
       -DBUILD_opencv_java=OFF \
       -DBUILD_SHARED_LIBS=OFF \
       -DWITH_CUDA=ON \
       -DBUILD_PERF_TESTS=OFF \
       -DBUILD_TESTS=OFF \
       # -DENABLE_FAST_MATH=1 \
       # -DCUDA_FAST_MATH=1 \
       -DWITH_CUBLAS=1 \
       -DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda \
       -DOPENCV_EXTRA_MODULES_PATH=../../opencv_contrib-4.0.1/modules/ \
       ##
       ## Should compile for most card
       ## 3.5 binary code for devices with compute capability 3.5 and 3.7,
       ## 5.0 binary code for devices with compute capability 5.0 and 5.2,
       ## 6.0 binary code for devices with compute capability 6.0 and 6.1,
       -DCUDA_ARCH_BIN='7.0' \
       -DCUDA_ARCH_PTX="" \
       ##
       ## AVX in dispatch because not all machines have it
       -DCPU_DISPATCH=AVX,AVX2 \
       -DENABLE_PRECOMPILED_HEADERS=OFF \
       -DWITH_OPENGL=OFF \
       -DWITH_OPENCL=OFF \
       -DWITH_QT=OFF \
       -DWITH_NVCUVID=ON \
       -DWITH_IPP=ON \
       -DWITH_TBB=ON \
       -DFORCE_VTK=ON \
       -DWITH_EIGEN=ON \
       -DWITH_V4L=ON \
       -DWITH_XINE=ON \
       -DWITH_GDAL=ON \
       -DWITH_1394=OFF \
       -DWITH_FFMPEG=OFF \
       -DBUILD_PROTOBUF=OFF \
       -DBUILD_TESTS=OFF \
       -DBUILD_PERF_TESTS=OFF \
       -DBUILD_opencv_xfeatures2d=OFF \
       -DCMAKE_BUILD_TYPE=RELEASE \
       -DCMAKE_INSTALL_PREFIX=/usr/local \
    .. \
    && make -j15 \
    && make install \
    && rm /4.0.1.zip \
    && rm /cmake-3.13.0.tar.gz
+21 −0
Original line number Diff line number Diff line
/*
 * Copyright 1993-2015 NVIDIA Corporation.  All rights reserved.
 *
 * Please refer to the NVIDIA end user license agreement (EULA) associated
 * with this source code for terms and conditions that govern your use of
 * this software. Any use, reproduction, disclosure, or distribution of
 * this software and related documentation outside the terms of the EULA
 * is strictly prohibited.
 *
 */
 
#ifndef __cuda_h__
#define __cuda_h__

/**
* CUDA API version support
*/

#include "dynlink_cuda_cuda.h"

#endif //__cuda_h__
+1689 −0

File added.

Preview size limit exceeded, changes collapsed.

+913 −0

File added.

Preview size limit exceeded, changes collapsed.

Loading