|
|
|
@ -130,9 +130,9 @@ RUN curl -s -q https://glide.sh/get | sh
|
|
|
|
|
# 2. Manually add ~IPluginFactory() in IPluginFactory class of NvInfer.h, otherwise, it couldn't work in paddle.
|
|
|
|
|
# See https://github.com/PaddlePaddle/Paddle/issues/10129 for details.
|
|
|
|
|
|
|
|
|
|
RUN wget -q https://paddle-ci.gz.bcebos.com/TensorRT-5.1.5.0.Ubuntu-14.04.5.x86_64-gnu.cuda-10.1.cudnn7.5.tar.gz --no-check-certificate && \
|
|
|
|
|
tar -zxf TensorRT-5.1.5.0.Ubuntu-14.04.5.x86_64-gnu.cuda-10.1.cudnn7.5.tar.gz -C /usr/local && \
|
|
|
|
|
cp -rf /usr/local/TensorRT-5.1.5.0/include /usr
|
|
|
|
|
RUN wget -q https://paddlepaddledeps.cdn.bcebos.com/TensorRT-6.0.1.5.Ubuntu-16.04.x86_64-gnu.cuda-10.1.cudnn7.6.tar.gz --no-check-certificate && \
|
|
|
|
|
tar -zxf TensorRT-6.0.1.5.Ubuntu-16.04.x86_64-gnu.cuda-10.1.cudnn7.6.tar.gz -C /usr/local && \
|
|
|
|
|
cp -rf /usr/local/TensorRT-6.0.1.5/include/* /usr/include/ && cp -rf /usr/local/TensorRT-6.0.1.5/lib/* /usr/lib/
|
|
|
|
|
|
|
|
|
|
# git credential to skip password typing
|
|
|
|
|
RUN git config --global credential.helper store
|
|
|
|
|