This commit is contained in:
Andrew Tulloch 2018-11-09 21:08:04 -08:00 коммит произвёл Tianqi Chen
Родитель fc83c7f26d
Коммит 3ee13fc5f2
2 изменённых файлов: 17 добавлений и 0 удалений

Просмотреть файл

@ -35,4 +35,8 @@ RUN bash /install/ubuntu_install_redis.sh
COPY install/ubuntu_install_golang.sh /install/ubuntu_install_golang.sh
RUN bash /install/ubuntu_install_golang.sh
# NNPACK deps
COPY install/ubuntu_install_nnpack.sh /install/ubuntu_install_nnpack.sh
RUN bash /install/ubuntu_install_nnpack.sh
ENV PATH $PATH:$CARGO_HOME/bin:/usr/lib/go-1.10/bin

Просмотреть файл

@ -0,0 +1,13 @@
apt-get update && apt-get install -y --no-install-recommends --force-yes git cmake
git clone https://github.com/Maratyszcza/NNPACK NNPACK
cd NNPACK
# TODO: specific tag?
git checkout 1e005b0c2
cd -
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 -