2019-02-19 21:18:08 +08:00
|
|
|
#!/usr/bin/env bash
|
|
|
|
|
|
|
|
number_of_build_workers=8
|
2019-03-29 00:54:27 +08:00
|
|
|
create_shared_lib="-DBUILD_SHARED_LIBS:BOOL=ON"
|
|
|
|
|
|
|
|
#my_cuda_compute_model=75 #Compute capability for Tesla T4, RTX 2080
|
|
|
|
#my_cuda_compute_model=72 #Compute capability for Jetson Xavier
|
|
|
|
#my_cuda_compute_model=70 #Compute capability for Tesla V100
|
|
|
|
#my_cuda_compute_model=62 #Compute capability for Jetson TX2
|
|
|
|
#my_cuda_compute_model=61 #Compute capability for Tesla P40
|
|
|
|
#my_cuda_compute_model=60 #Compute capability for Tesla P100
|
|
|
|
#my_cuda_compute_model=53 #Compute capability for Jetson TX1
|
|
|
|
#my_cuda_compute_model=52 #Compute capability for Tesla M40/M60
|
|
|
|
#my_cuda_compute_model=37 #Compute capability for Tesla K80
|
|
|
|
#my_cuda_compute_model=35 #Compute capability for Tesla K20/K40
|
|
|
|
#my_cuda_compute_model=30 #Compute capability for Tesla K10, Quadro K4000
|
2019-02-19 21:18:08 +08:00
|
|
|
|
2019-03-04 22:15:19 +08:00
|
|
|
if [[ "$OSTYPE" == "darwin"* ]]; then
|
|
|
|
OpenCV_DIR="/usr/local/Cellar/opencv@3/3.4.5"
|
|
|
|
additional_defines="-DOpenCV_DIR=${OpenCV_DIR}"
|
|
|
|
if [[ "$1" == "gcc" ]]; then
|
|
|
|
export CC="/usr/local/bin/gcc-8"
|
|
|
|
export CXX="/usr/local/bin/g++-8"
|
|
|
|
fi
|
|
|
|
fi
|
|
|
|
|
2019-03-29 00:54:27 +08:00
|
|
|
if [[ ! -z "$my_cuda_compute_model" ]]; then
|
|
|
|
additional_build_setup="-DCUDA_COMPUTE_MODEL=${my_cuda_compute_model}"
|
|
|
|
fi
|
2019-03-04 22:15:19 +08:00
|
|
|
|
2019-02-20 18:42:53 +08:00
|
|
|
# RELEASE
|
2019-02-19 21:18:08 +08:00
|
|
|
mkdir -p build_release
|
|
|
|
cd build_release
|
2019-03-29 00:54:27 +08:00
|
|
|
cmake .. -DCMAKE_BUILD_TYPE=Release ${additional_defines} ${create_shared_lib} ${additional_build_setup}
|
2019-02-19 21:18:08 +08:00
|
|
|
cmake --build . --target install -- -j${number_of_build_workers}
|
|
|
|
#cmake --build . --target install --parallel ${number_of_build_workers} #valid only for CMake 3.12+
|
2019-03-14 05:56:44 +08:00
|
|
|
rm -f DarknetConfig.cmake
|
|
|
|
rm -f DarknetConfigVersion.cmake
|
2019-02-19 21:18:08 +08:00
|
|
|
cd ..
|
2019-03-14 05:56:44 +08:00
|
|
|
cp cmake/Modules/*.cmake share/darknet
|
2019-03-04 06:35:38 +08:00
|
|
|
|
|
|
|
# DEBUG
|
|
|
|
mkdir -p build_debug
|
|
|
|
cd build_debug
|
2019-03-29 00:54:27 +08:00
|
|
|
cmake .. -DCMAKE_BUILD_TYPE=Debug ${additional_defines} ${create_shared_lib} ${additional_build_setup}
|
2019-03-04 06:35:38 +08:00
|
|
|
cmake --build . --target install -- -j${number_of_build_workers}
|
|
|
|
#cmake --build . --target install --parallel ${number_of_build_workers} #valid only for CMake 3.12+
|
2019-03-14 05:56:44 +08:00
|
|
|
rm -f DarknetConfig.cmake
|
|
|
|
rm -f DarknetConfigVersion.cmake
|
2019-03-04 06:35:38 +08:00
|
|
|
cd ..
|
2019-03-14 05:56:44 +08:00
|
|
|
cp cmake/Modules/*.cmake share/darknet
|