From 6095814229b2354d4445bd7083e7d13d37f772aa Mon Sep 17 00:00:00 2001 From: Junjie Bai Date: Fri, 29 Mar 2019 22:42:18 -0700 Subject: [PATCH] Fix trt installation in CI (#18609) Summary: caffe2_py2_cuda9_0_cudnn7_ubuntu16_04_build is failing ``` ... Mar 29 04:44:46 Need to get 174 MB of archives. Mar 29 04:44:46 After this operation, 576 MB of additional disk space will be used. Mar 29 04:44:46 Do you want to continue? [Y/n] Abort. Exited with code 1 ... ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/18609 Differential Revision: D14694990 Pulled By: bddppq fbshipit-source-id: 260446a8650f660a2baf123a3f17efdf0a8d6c64 --- .jenkins/caffe2/build.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.jenkins/caffe2/build.sh b/.jenkins/caffe2/build.sh index 7e8fdea..0e56887 100755 --- a/.jenkins/caffe2/build.sh +++ b/.jenkins/caffe2/build.sh @@ -142,7 +142,7 @@ if [[ $BUILD_ENVIRONMENT == *py2-cuda9.0-cudnn7-ubuntu16.04* ]]; then sudo dpkg -i ./nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda9.0_1-1_amd64.deb sudo apt-key add /var/nvinfer-runtime-trt-repo-5.0.2-ga-cuda9.0/7fa2af80.pub sudo apt-get -qq update - sudo apt-get install libnvinfer5 libnvinfer-dev + sudo apt-get install -y --no-install-recommends libnvinfer5=5.0.2-1+cuda9.0 libnvinfer-dev=5.0.2-1+cuda9.0 rm ./nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda9.0_1-1_amd64.deb build_args+=("USE_TENSORRT=ON") -- 2.7.4