Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Pinoxchio
apollo
提交
6712cb24
A
apollo
项目概览
Pinoxchio
/
apollo
与 Fork 源项目一致
从无法访问的项目Fork
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
A
apollo
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
6712cb24
编写于
9月 29, 2020
作者:
S
storypku
提交者:
Chang Songhong
9月 29, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Docker: gpu support adjustment
上级
d7791aec
变更
4
显示空白变更内容
内联
并排
Showing
4 changed file
with
9 addition
and
121 deletion
+9
-121
docker/build/installers/install_gpu_support.sh
docker/build/installers/install_gpu_support.sh
+1
-10
docker/build/installers/install_jetson_ml.sh
docker/build/installers/install_jetson_ml.sh
+0
-1
docker/build/installers/install_tensorrt.sh
docker/build/installers/install_tensorrt.sh
+0
-110
docker/build/pre_cyber.x86_64.dockerfile
docker/build/pre_cyber.x86_64.dockerfile
+8
-0
未找到文件。
docker/build/installers/install_gpu_support.sh
浏览文件 @
6712cb24
...
...
@@ -30,16 +30,7 @@ apt_get_update_and_install \
TARGET_ARCH
=
"
$(
uname
-m
)
"
# Note(storypku):
# 1) cublas disabled for x86 since already installed before this installer runs
if
[[
"
${
TARGET_ARCH
}
"
==
"aarch64"
]]
;
then
apt_get_update_and_install
\
libcublas10
\
libcublas-dev
info
"Install TensorRT 7 ..."
bash
${
CURR_DIR
}
/install_tensorrt.sh
fi
# 1) cublas was already installed before this installer runs
info
"Install libtorch ..."
bash
${
CURR_DIR
}
/install_libtorch.sh
...
...
docker/build/installers/install_jetson_ml.sh
浏览文件 @
6712cb24
...
...
@@ -95,7 +95,6 @@ TRT_VER_APT="${TRT_VERSION}-1+cuda${CUDA_VER}"
TRT_PKGS
=(
libnvinfer
${
TRT_MAJOR
}
_
${
TRT_VER_APT
}
_arm64.deb
libnvinfer-bin_
${
TRT_VER_APT
}
_arm64.deb
libnvinfer-dev_
${
TRT_VER_APT
}
_arm64.deb
libnvinfer-plugin
${
TRT_MAJOR
}
_
${
TRT_VER_APT
}
_arm64.deb
libnvinfer-plugin-dev_
${
TRT_VER_APT
}
_arm64.deb
...
...
docker/build/installers/install_tensorrt.sh
已删除
100755 → 0
浏览文件 @
d7791aec
#!/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]
}
"
)
"
.
./installer_base.sh
ARCH
=
"
$(
uname
-m
)
"
if
[
"
${
ARCH
}
"
=
"aarch64"
]
;
then
CUDA_VER
=
"10.2"
TRT_VER1
=
"7.1.0-1"
MAJOR
=
"
${
TRT_VER1
%%.*
}
"
TRT_VERSION
=
"
${
TRT_VER1
}
+cuda
${
CUDA_VER
}
"
TRT_PKGS
=
"
\
libnvinfer
${
MAJOR
}
_
${
TRT_VERSION
}
_arm64.deb
\
libnvinfer-bin_
${
TRT_VERSION
}
_arm64.deb
\
libnvinfer-dev_
${
TRT_VERSION
}
_arm64.deb
\
libnvinfer-plugin
${
MAJOR
}
_
${
TRT_VERSION
}
_arm64.deb
\
libnvinfer-plugin-dev_
${
TRT_VERSION
}
_arm64.deb
\
libnvonnxparsers
${
MAJOR
}
_
${
TRT_VERSION
}
_arm64.deb
\
libnvonnxparsers-dev_
${
TRT_VERSION
}
_arm64.deb
\
libnvparsers
${
MAJOR
}
_
${
TRT_VERSION
}
_arm64.deb
\
libnvparsers-dev_
${
TRT_VERSION
}
_arm64.deb
\
"
# tensorrt_7.1.0.16-1+cuda10.2_arm64.deb
# libnvinfer-doc_${TRT_VERSION}_all.deb
# libnvinfer-samples_${TRT_VERSION}_all.deb
for
pkg
in
${
TRT_PKGS
}
;
do
info
"Downloading
${
LOCAL_HTTP_ADDR
}
/
${
pkg
}
"
wget
"
${
LOCAL_HTTP_ADDR
}
/
${
pkg
}
"
>
/dev/null
done
dpkg
-i
${
TRT_PKGS
}
info
"Successfully installed TensorRT
${
MAJOR
}
"
rm
-rf
${
TRT_PKGS
}
apt-get clean
exit
0
fi
#Install the TensorRT package that fits your particular needs.
#For only running TensorRT C++ applications:
#sudo apt-get install libnvinfer7 libnvonnxparsers7 libnvparsers7 libnvinfer-plugin7
#For also building TensorRT C++ applications:
#sudo apt-get install libnvinfer-dev libnvonnxparsers-dev
# libnvparsers-dev libnvinfer-plugin-dev
#For running TensorRT Python applications:
#sudo apt-get install python-libnvinfer python3-libnvinfer
fixed_version
=
true
if
${
fixed_version
}
;
then
VERSION
=
"7.0.0-1+cuda10.2"
apt_get_update_and_install
\
libnvinfer7
=
"
${
VERSION
}
"
\
libnvonnxparsers7
=
"
${
VERSION
}
"
\
libnvparsers7
=
"
${
VERSION
}
"
\
libnvinfer-plugin7
=
"
${
VERSION
}
"
\
libnvinfer-dev
=
"
${
VERSION
}
"
\
libnvonnxparsers-dev
=
"
${
VERSION
}
"
\
libnvparsers-dev
=
"
${
VERSION
}
"
\
libnvinfer-plugin-dev
=
"
${
VERSION
}
"
else
apt_get_update_and_install
\
libnvinfer7
\
libnvonnxparsers7
\
libnvparsers7
\
libnvinfer-plugin7
\
libnvinfer-dev
\
libnvonnxparsers-dev
\
libnvparsers-dev
\
libnvinfer-plugin-dev
fi
# FIXME(all):
# Previously soft sym-linked for successful caffe-1.0 compilation.
# Now that caffe-1.0 is retired, do we still need this?
# Minor changes required:
# 1) cudnn major version hard-code fix: v7,v8,...
# 2) move to cudnn installer section
CUDNN_HEADER_DIR
=
"/usr/include/
$(
uname
-m
)
-linux-gnu"
[
-e
"
${
CUDNN_HEADER_DIR
}
/cudnn.h"
]
||
ln
-s
"
${
CUDNN_HEADER_DIR
}
/cudnn_v7.h"
"
${
CUDNN_HEADER_DIR
}
/cudnn.h"
# Disable nvidia apt sources.list settings to speed up build process
rm
-f
/etc/apt/sources.list.d/nvidia-ml.list
rm
-f
/etc/apt/sources.list.d/cuda.list
docker/build/pre_cyber.x86_64.dockerfile
浏览文件 @
6712cb24
...
...
@@ -3,6 +3,14 @@ FROM ${BASE_IMAGE}
ENV
TENSORRT_VERSION 7.0.0-1+cuda10.2
#For only running TensorRT C++ applications:
# sudo apt-get install libnvinfer7 libnvonnxparsers7 libnvparsers7 libnvinfer-plugin7
#For also building TensorRT C++ applications:
# sudo apt-get install libnvinfer-dev libnvonnxparsers-dev \
# libnvparsers-dev libnvinfer-plugin-dev
#For running TensorRT Python applications:
# sudo apt-get install python3-libnvinfer
RUN
apt-get
-y
update
\
&&
apt-get
install
-y
--no-install-recommends
\
libnvinfer7
=
"
${
TENSORRT_VERSION
}
"
\
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录