[DOCKER] Update ci-gpu to include NNPack (#2846)

a1c2fd1b · Tianqi Chen · GitHub · af4efdbf · a1c2fd1b · a1c2fd1b
Unverified Commit a1c2fd1b authored Mar 18, 2019 by Tianqi Chen Committed by GitHub Mar 18, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletions

Jenkinsfile
+1 -1

docker/Dockerfile.ci_gpu
+4 -0

No files found.
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -21,7 +21,7 @@
 // - Periodically cleanup the old versions on local workers
 //
 ci_lint = "tvmai/ci-lint:v0.50"
-ci_gpu = "tvmai/ci-gpu:v0.50"
+ci_gpu = "tvmai/ci-gpu:v0.51"
 ci_cpu = "tvmai/ci-cpu:v0.41"
 ci_i386 = "tvmai/ci-i386:v0.50"


--- a/docker/Dockerfile.ci_gpu
+++ b/docker/Dockerfile.ci_gpu
@@ -82,6 +82,10 @@ RUN bash /install/ubuntu_install_redis.sh
 COPY install/ubuntu_install_antlr.sh /install/ubuntu_install_antlr.sh
 RUN bash /install/ubuntu_install_antlr.sh

+# NNPACK deps
+COPY install/ubuntu_install_nnpack.sh /install/ubuntu_install_nnpack.sh
+RUN bash /install/ubuntu_install_nnpack.sh
+
 # Environment variables
 ENV PATH=/usr/local/nvidia/bin:${PATH}
 ENV PATH=/usr/local/cuda/bin:${PATH}