Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
T
tensorflow-feed-from-gpu
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Commits
Issue Boards
Open sidebar
Elphel
tensorflow-feed-from-gpu
Commits
367edb17
Commit
367edb17
authored
Mar 11, 2019
by
fierval
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
current state
parent
9bc751e7
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
27 additions
and
25 deletions
+27
-25
CMakeLists.txt
CMakeLists.txt
+2
-2
Dockerfile
docker/Dockerfile
+25
-23
No files found.
CMakeLists.txt
View file @
367edb17
...
...
@@ -44,9 +44,9 @@ include_directories("/usr/local/tensorflow/include/")
# IMPORTANT: Protobuf includes. Depends on the anaconda path
# This is Azure DLVM (not sure if DSVM is the same)
include_directories
(
"/data/anaconda/envs/py36/lib/python3.6/site-packages/tensorflow/include/"
)
#
include_directories("/data/anaconda/envs/py36/lib/python3.6/site-packages/tensorflow/include/")
# This is a standard install of Anaconda with p36 environment
include_directories
(
"
${
MYHOME
}
/anaconda3/envs/py36/lib/python3.6/site-packages/tensorflow/include/"
)
#
include_directories("${MYHOME}/anaconda3/envs/py36/lib/python3.6/site-packages/tensorflow/include/")
target_link_libraries
(
tf_detector_example
${
CUDA_LIBRARIES
}
...
...
docker/Dockerfile
View file @
367edb17
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
\
RUN
apt-get
install
-y
git libgtk2.0-dev curl pkg-config
autoconf automake libtool
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
...
...
@@ -24,21 +21,22 @@ RUN tar xzvf /cmake-3.13.0.tar.gz -C / \
&&
make
-j15
\
&&
make
install
# Second: get and build OpenCV
4.0
.1
# Second: get and build OpenCV
3.3
.1
#
ADD
https://github.com/protocolbuffers/protobuf/releases/download/v3.6.1/protobuf-cpp-3.6.1.tar.gz /
RUN
tar
xzvf /protobuf-cpp-3.6.1.tar.gz
-C
/
RUN
cd
/protobuf-3.6.1
\
&&
./configure
\
&&
make
-j15
\
&&
make
install
\
&&
ldconfig
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_contrib/archive/3.3.1.zip /
RUN
unzip
-o
/3.3.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
\
ADD
https://github.com/opencv/opencv/archive/
3.3
.1.zip /
RUN
unzip
-o
/
3.3
.1.zip
RUN
cd
/workspace/opencv-
3.3
.1
\
&&
mkdir
build
\
&&
cd
build
\
&&
cmake
-DBUILD_TIFF
=
ON
\
...
...
@@ -47,16 +45,13 @@ RUN cd /workspace/opencv-4.0.1 \
-DWITH_CUDA
=
ON
\
-DBUILD_PERF_TESTS
=
OFF
\
-DBUILD_TESTS
=
OFF
\
-DBUILD_opencv_codacodec
=
ON
\
# -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/
\
-DOPENCV_EXTRA_MODULES_PATH
=
../../opencv_contrib-
3.3
.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="" \
##
...
...
@@ -76,7 +71,7 @@ RUN cd /workspace/opencv-4.0.1 \
-DWITH_GDAL=ON \
-DWITH_1394=OFF \
-DWITH_FFMPEG=OFF \
-DBUILD_PROTOBUF=O
FF
\
-DBUILD_PROTOBUF=O
N
\
-DBUILD_TESTS=OFF \
-DBUILD_PERF_TESTS=OFF \
-DBUILD_opencv_xfeatures2d=OFF \
...
...
@@ -85,5 +80,12 @@ RUN cd /workspace/opencv-4.0.1 \
.. \
&& make -j15 \
&& make install \
&& rm /
4.0
.1.zip \
&& rm /
3.3
.1.zip \
&& rm /cmake-3.13.0.tar.gz
RUN
mkdir
fast_od
RUN
mkdir
/home/boris
# tensorflow libraries
ADD
tensorflow.tar /
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment