diff --git a/Jenkinsfile b/Jenkinsfile index d68d548d8abf..a56296c98c54 100644 --- 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" diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index fc6a84a6d7f7..4c1ffb14c1f2 100644 --- 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}