|
|
@ -53,7 +53,7 @@ RUN curl -s -q https://glide.sh/get | sh
|
|
|
|
# and its size is only one-third of the official one.
|
|
|
|
# and its size is only one-third of the official one.
|
|
|
|
# 2. Manually add ~IPluginFactory() in IPluginFactory class of NvInfer.h, otherwise, it couldn't work in paddle.
|
|
|
|
# 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.
|
|
|
|
# See https://github.com/PaddlePaddle/Paddle/issues/10129 for details.
|
|
|
|
RUN wget -qO- http://paddlepaddledeps.bj.bcebos.com/TensorRT-4.0.0.3.Ubuntu-16.04.4.x86_64-gnu.cuda-8.0.cudnn7.0.tar.gz | \
|
|
|
|
RUN wget -qO- http://paddlepaddledeps.cdn.bcebos.com/TensorRT-4.0.0.3.Ubuntu-16.04.4.x86_64-gnu.cuda-8.0.cudnn7.0.tar.gz | \
|
|
|
|
tar -xz -C /usr/local && \
|
|
|
|
tar -xz -C /usr/local && \
|
|
|
|
cp -rf /usr/local/TensorRT/include /usr && \
|
|
|
|
cp -rf /usr/local/TensorRT/include /usr && \
|
|
|
|
cp -rf /usr/local/TensorRT/lib /usr
|
|
|
|
cp -rf /usr/local/TensorRT/lib /usr
|
|
|
|