ubuntu_install_nnpack.sh 462 Bytes
Newer Older
1
#!/bin/bash
2

3 4 5 6 7
set -e
set -u
set -o pipefail

apt-get update && apt-get install -y --no-install-recommends git cmake
8 9

# TODO: specific tag?
10 11
git clone https://github.com/Maratyszcza/NNPACK NNPACK
(cd NNPACK && git checkout 1e005b0c2)
12 13 14 15 16

mkdir -p NNPACK/build
cd NNPACK/build
cmake -DCMAKE_INSTALL_PREFIX:PATH=. -DNNPACK_INFERENCE_ONLY=OFF -DNNPACK_CONVOLUTION_ONLY=OFF -DNNPACK_BUILD_TESTS=OFF -DCMAKE_POSITION_INDEPENDENT_CODE=ON .. && make -j4 && make install
cd -