diff --git a/.travis.yml b/.travis.yml index c96c9d5f279fdbd310d81a8e6eee1544259dbbd3..13c8ca0ca93a64d953f103ed1a019af1c9fc674e 100644 --- a/.travis.yml +++ b/.travis.yml @@ -26,7 +26,7 @@ env: # WITH_CUDNN: true - BUILD_NAME="default-cmake-cuda8" # - BUILD_NAME="default-make-cuda8" WITH_CMAKE=false - # - BUILD_NAME="default-cmake-cpu" WITH_CUDA=false + - BUILD_NAME="default-cmake-cpu" WITH_CUDA=false # - BUILD_NAME="python3-cmake-cuda8" WITH_PYTHON3=true # - BUILD_NAME="default-make-cuda8_nocudnn" WITH_CUDNN=false @@ -45,18 +45,19 @@ install: - sudo bash ./3rdparty/ubuntu/install_ubuntu_deps_and_cuda.sh - sudo apt-get -y install libatlas-base-dev - sudo apt-get -y install libopencv-dev - - pip install --upgrade numpy # Running CMake before_script: - - mkdir build - - cd build - - cmake -DBUILD_CAFFE=ON -DDOWNLOAD_BODY_25_MODEL=OFF -DDOWNLOAD_BODY_COCO_MODEL=OFF -DDOWNLOAD_FACE_MODEL=OFF -DDOWNLOAD_HAND_MODEL=OFF -DDOWNLOAD_BODY_MPI_MODEL=OFF .. + - bash ./cmake/travis/configure.sh + # Equivalent for CMake + # - mkdir build + # - cd build + # - cmake -DBUILD_CAFFE=ON -DDOWNLOAD_BODY_25_MODEL=OFF -DDOWNLOAD_BODY_COCO_MODEL=OFF -DDOWNLOAD_FACE_MODEL=OFF -DDOWNLOAD_HAND_MODEL=OFF -DDOWNLOAD_BODY_MPI_MODEL=OFF .. # Build your code e.g., by calling make script: - - make -j`nproc` - # - bash ./cmake/travis/build.sh + - bash ./cmake/travis/run_make.sh + # - make -j`nproc` # Equivalent to make -j`nproc` # - no_cores=`cat /proc/cpuinfo | grep processor | wc -l` # - make -j${no_cores} diff --git a/3rdparty/ubuntu/install_ubuntu_deps_and_cuda.sh b/3rdparty/ubuntu/install_ubuntu_deps_and_cuda.sh index e4254dca6d34562478c84a0d8abe514b85c177a2..4d9d16074e592b13d427b522521595024dda1c8c 100644 --- a/3rdparty/ubuntu/install_ubuntu_deps_and_cuda.sh +++ b/3rdparty/ubuntu/install_ubuntu_deps_and_cuda.sh @@ -3,10 +3,10 @@ ### INSTALL PREREQUISITES # Install CUDA 8.0 -bash install_cuda.sh +bash 3rdparty/ubuntu/install_cuda.sh # Install cuDNN 5.1 -bash install_cudnn.sh +bash 3rdparty/ubuntu/install_cudnn.sh # Caffe prerequisites -bash install_ubuntu_deps.sh +bash 3rdparty/ubuntu/install_ubuntu_deps.sh diff --git a/cmake/travis/configure-cmake.sh b/cmake/travis/configure-cmake.sh new file mode 100644 index 0000000000000000000000000000000000000000..e093d450b0717d1bdb1f357c1edb0d748d3415cc --- /dev/null +++ b/cmake/travis/configure-cmake.sh @@ -0,0 +1,28 @@ +# CMake configuration + +mkdir build +# mkdir -p build +cd build + +ARGS="-DDOWNLOAD_BODY_25_MODEL=OFF -DDOWNLOAD_FACE_MODEL=OFF -DDOWNLOAD_HAND_MODEL=OFF" +# ARGS="-DBUILD_CAFFE=ON -DDOWNLOAD_BODY_25_MODEL=OFF -DDOWNLOAD_BODY_COCO_MODEL=OFF -DDOWNLOAD_FACE_MODEL=OFF -DDOWNLOAD_HAND_MODEL=OFF -DDOWNLOAD_BODY_MPI_MODEL=OFF" + +if $WITH_PYTHON3 ; then + ARGS="$ARGS -DBUILD_PYTHON=On" + # ARGS="$ARGS -Dpython_version=3" +fi + +if $WITH_CUDA ; then + # Only build SM50 + ARGS="$ARGS -DGPU_MODE=CUDA -DCUDA_ARCH=Manual -DCUDA_ARCH_BIN=\"52\" -DCUDA_ARCH_PTX=\"50\"" +else + ARGS="$ARGS -DGPU_MODE=CPU_ONLY -DUSE_MKL=Off" +fi + +if $WITH_CUDNN ; then + ARGS="$ARGS -DUSE_CUDNN=On" +else + ARGS="$ARGS -DUSE_CUDNN=Off" +fi + +cmake .. $ARGS diff --git a/cmake/travis/configure.sh b/cmake/travis/configure.sh new file mode 100644 index 0000000000000000000000000000000000000000..ef740c8982e74e18cf6f290f19156642586fbe92 --- /dev/null +++ b/cmake/travis/configure.sh @@ -0,0 +1,11 @@ +#!/bin/bash +# configure the project + +BASEDIR=$(dirname $0) +source $BASEDIR/defaults.sh + +if ! $WITH_CMAKE ; then + source $BASEDIR/configure-make.sh +else + source $BASEDIR/configure-cmake.sh +fi diff --git a/cmake/travis/run_make.sh b/cmake/travis/run_make.sh new file mode 100644 index 0000000000000000000000000000000000000000..f4862e6052dc243f9b158ef2c47e576b6ba8c55b --- /dev/null +++ b/cmake/travis/run_make.sh @@ -0,0 +1,14 @@ +#!/bin/bash +# build the project + +BASEDIR=$(dirname $0) +source $BASEDIR/defaults.sh + +if $WITH_CMAKE ; then + cd build + make -j`nproc` + # make --jobs $NUM_THREADS +else # if ! $WITH_CMAKE ; then + make all -j`nproc` + # make --jobs $NUM_THREADS all +fi