Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
75e9a6d1
P
Paddle
项目概览
机器未来
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
75e9a6d1
编写于
7月 27, 2021
作者:
W
Wilber
提交者:
GitHub
7月 27, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update compile images for trt6.0.1.8 (#34341)
上级
0dd6a44a
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
4 addition
and
1 deletion
+4
-1
tools/dockerfile/build_scripts/install_trt.sh
tools/dockerfile/build_scripts/install_trt.sh
+3
-0
tools/dockerfile/centos7_manylinux.sh
tools/dockerfile/centos7_manylinux.sh
+1
-1
未找到文件。
tools/dockerfile/build_scripts/install_trt.sh
浏览文件 @
75e9a6d1
...
...
@@ -51,6 +51,9 @@ elif [[ "$VERSION" == "10.2" && "$CUDNN_VERSION" == "7.6.5" ]];then
tar
-zxf
TensorRT-6.0.1.8.CentOS-7.6.x86_64-gnu.cuda-10.2.cudnn7.6.tar.gz
-C
/usr/local
cp
-rf
/usr/local/TensorRT-6.0.1.8/include/
*
/usr/include/
&&
cp
-rf
/usr/local/TensorRT-6.0.1.8/lib/
*
/usr/lib/
rm
-f
TensorRT-6.0.1.8.CentOS-7.6.x86_64-gnu.cuda-10.2.cudnn7.6.tar.gz
# trt6.0.1.8 should hack some code.
sed
-i
"s/virtual int getConstantValue() const = 0;/&
\n
protected:
\n
virtual ~IDimensionExpr() {};/g"
/usr/local/TensorRT-6.0.1.8/include/NvInferRuntime.h
sed
-i
"s/virtual IPlugin
\*
createPlugin(const char
\*
layerName, const void
\*
serialData, size_t serialLength) TRTNOEXCEPT = 0;/&
\n
protected:
\n
virtual ~IPluginFactory() {}/g"
/usr/local/TensorRT-6.0.1.8/include/NvInferRuntime.h
elif
[[
"
$VERSION
"
==
"10.2"
&&
"
$CUDNN_VERSION
"
==
"8.1.1"
]]
;
then
wget https://paddle-ci.gz.bcebos.com/TRT/TensorRT-7.2.3.4.CentOS-7.9.x86_64-gnu.cuda-10.2.cudnn8.1.tar.gz
--no-check-certificate
tar
-zxf
TensorRT-7.2.3.4.CentOS-7.9.x86_64-gnu.cuda-10.2.cudnn8.1.tar.gz
-C
/usr/local
...
...
tools/dockerfile/centos7_manylinux.sh
浏览文件 @
75e9a6d1
...
...
@@ -44,7 +44,7 @@ function make_cuda102cudnn7() {
function
make_cuda102cudnn7gcc54
()
{
sed
's/<baseimg>/10.2-cudnn7-devel-centos7/g'
Dockerfile.centos
>
Dockerfile.tmp
sed
-i
"s#RUN bash build_scripts/build.sh#RUN bash build_scripts/install_gcc.sh gcc54
\n
RUN mv /usr/bin/cc /usr/bin/cc.bak
\&\&
ln -s /usr/local/gcc-
8.2/bin/gcc /usr/bin/cc
\n
ENV PATH=/usr/local/gcc-8.2
/bin:
\$
PATH
\n
RUN bash build_scripts/build.sh#g"
Dockerfile.tmp
sed
-i
"s#RUN bash build_scripts/build.sh#RUN bash build_scripts/install_gcc.sh gcc54
\n
RUN mv /usr/bin/cc /usr/bin/cc.bak
\&\&
ln -s /usr/local/gcc-
5.4/bin/gcc /usr/bin/cc
\n
ENV PATH=/usr/local/gcc-5.4
/bin:
\$
PATH
\n
RUN bash build_scripts/build.sh#g"
Dockerfile.tmp
}
function
make_cuda102cudnn8
()
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录