提交 dbfc8690 编写于 作者: L Liu Jiaming 提交者: Xiangquan Xiao

TensorRT version bumped from 5.x to 7.0.0.11

上级 5c728fd2
#!/usr/bin/env bash
###############################################################################
# Copyright 2020 The Apollo Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
###############################################################################
# Fail on first error.
set -e
cd "$(dirname "${BASH_SOURCE[0]}")"
VERSION="7.0.0.11"
PKG_NAME="TensorRT-${VERSION}.Ubuntu-18.04.x86_64-gnu.cuda-10.0.cudnn7.6.tar.gz"
# Note(storypku)
# Download the TensorRT7 tarball from nvidia.com, and start the http
# server right at the archive directory:
# python3 -m http.server 8081
DOWNLOAD_LINK="http://my_host_ip:8081/${PKG_NAME}"
wget "${DOWNLOAD_LINK}"
tar xzf "${PKG_NAME}"
pushd /usr/lib/x86_64-linux-gnu
rm -rf libnvcaffe_parser* libnvinfer* libnvonnxparser.so* libnvparsers*
popd
rm -rf /usr/include/tensorrt
pushd TensorRT-${VERSION}
mv include /usr/include/tensorrt
mv lib/* /usr/lib/x86_64-linux-gnu/
popd
rm -rf ${PKG_NAME} TensorRT-${VERSION}
FROM apolloauto/apollo:dev-18.04-x86_64-20200428_2300
ENV DEBIAN_FRONTEND=noninteractive
COPY installers /tmp/installers
RUN sudo bash /tmp/installers/install_tensorrt_temp.sh
WORKDIR /apollo
USER apollo
......@@ -21,7 +21,7 @@ FAST_BUILD_MODE="no"
FAST_TEST_MODE="no"
VERSION=""
ARCH=$(uname -m)
VERSION_X86_64="dev-18.04-x86_64-20200428_2300"
VERSION_X86_64="dev-18.04-x86_64-20200507_0633"
VERSION_AARCH64="dev-aarch64-20170927_1111"
VERSION_OPT=""
NO_PULL_IMAGE=""
......
......@@ -29,6 +29,7 @@ cc_library(
},
no_match_error = "Please Build with an ARM or Linux x86_64 platform",
) + [
"-lmyelin",
"-lnvcaffe_parser",
"-lnvinfer",
"-lnvinfer_plugin",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册