Skip to content

Instantly share code, notes, and snippets.

@red010182
Last active July 24, 2017 06:04
Show Gist options
  • Save red010182/cb8133c1bd4daf41c82803d819ce1784 to your computer and use it in GitHub Desktop.
Save red010182/cb8133c1bd4daf41c82803d819ce1784 to your computer and use it in GitHub Desktop.
Install opencv in official tensorflow docker: gcr.io/tensorflow/tensorflow:latest-gpu
#!/bin/bash
apt-get -y update
apt-get -y upgrade
apt-get -y install build-essential cmake pkg-config git
apt-get -y install libjpeg-dev libtiff5-dev libjasper-dev libpng12-dev
apt-get -y install libavcodec-dev libavformat-dev libswscale-dev libv4l-dev libxvidcore-dev libx264-dev
apt-get -y install libgtk2.0-dev
apt-get -y install libtbb-dev
apt-get -y install libatlas-base-dev gfortran
cd /
git clone https://github.com/Itseez/opencv.git
cd opencv
git checkout 3.2.0
cd /
git clone https://github.com/Itseez/opencv_contrib.git
cd opencv_contrib
git checkout 3.2.0
export CUDA_CUDA_LIBRARY=/usr/local/cuda/lib64/
cd /opencv
mkdir build
cd build
cmake -DCUDA_CUDA_LIBRARY=/usr/local/cuda/lib64/stubs/libcuda.so\
-D CMAKE_BUILD_TYPE=RELEASE \
-D CMAKE_INSTALL_PREFIX=/usr/local \
-D INSTALL_C_EXAMPLES=OFF \
-D INSTALL_PYTHON_EXAMPLES=ON \
-D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules \
-D BUILD_EXAMPLES=ON ..
make -j8
make install
ldconfig
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment