Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
wmsofts
Paddle
提交
cf0d6027
P
Paddle
项目概览
wmsofts
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
cf0d6027
编写于
6月 18, 2022
作者:
W
Wilber
提交者:
GitHub
6月 18, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add cuda11.6 cudnn8.4 trt8.4 image (#43396)
上级
4a08c781
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
16 addition
and
0 deletion
+16
-0
tools/dockerfile/build_scripts/install_trt.sh
tools/dockerfile/build_scripts/install_trt.sh
+5
-0
tools/dockerfile/centos7_manylinux.sh
tools/dockerfile/centos7_manylinux.sh
+11
-0
未找到文件。
tools/dockerfile/build_scripts/install_trt.sh
浏览文件 @
cf0d6027
...
...
@@ -36,6 +36,11 @@ elif [[ "$1" == "trt8034" && "$VERSION" == "11.2" ]];then
tar
-zxf
TensorRT-8.0.3.4.Linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz
-C
/usr/local
cp
-rf
/usr/local/TensorRT-8.0.3.4/include/
*
/usr/include/
&&
cp
-rf
/usr/local/TensorRT-8.0.3.4/lib/
*
/usr/lib/
rm
TensorRT-8.0.3.4.Linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz
elif
[[
"
$1
"
==
"trt8406"
]]
;
then
wget
-q
https://paddle-ci.gz.bcebos.com/TRT/TensorRT-8.4.0.6.Linux.x86_64-gnu.cuda-11.6.cudnn8.3.tar.gz
--no-check-certificate
tar
-zxf
TensorRT-8.4.0.6.Linux.x86_64-gnu.cuda-11.6.cudnn8.3.tar.gz
-C
/usr/local
cp
-rf
/usr/local/TensorRT-8.4.0.6/include/
*
/usr/include/
&&
cp
-rf
/usr/local/TensorRT-8.4.0.6/lib/
*
/usr/lib/
rm
-f
TensorRT-8.4.0.6.Linux.x86_64-gnu.cuda-11.6.cudnn8.3.tar.gz
elif
[[
"
$VERSION
"
==
"11.2"
]]
;
then
wget
-q
https://paddle-ci.gz.bcebos.com/TRT/TensorRT7-cuda11.1-cudnn8.1.tar.gz
--no-check-certificate
tar
-zxf
TensorRT7-cuda11.1-cudnn8.1.tar.gz
-C
/usr/local
...
...
tools/dockerfile/centos7_manylinux.sh
浏览文件 @
cf0d6027
...
...
@@ -120,6 +120,14 @@ function make_cuda116cudnn8() {
sed
-i
"s#RUN bash build_scripts/build.sh#RUN bash build_scripts/install_gcc.sh gcc82
\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
}
function
make_cuda116cudnn840trt8406gcc82
()
{
sed
's/<baseimg>/11.6.2-cudnn8-devel-centos7/g'
Dockerfile.centos
>
Dockerfile.tmp
sed
-i
"s#RUN bash build_scripts/build.sh#RUN bash build_scripts/install_gcc.sh gcc82
\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
"/install_trt.sh/d"
Dockerfile.tmp
sed
-i
"s#RUN bash build_scripts/build.sh#RUN bash build_scripts/install_trt.sh trt8406
\n
RUN bash build_scripts/build.sh#g"
Dockerfile.tmp
sed
-i
'/CMD/iRUN ldconfig'
Dockerfile.tmp
}
function
main
()
{
local
CMD
=
$1
case
$CMD
in
...
...
@@ -177,6 +185,9 @@ function main() {
cuda116cudnn8
)
make_cuda116cudnn8
;;
cuda116cudnn840trt8406gcc82
)
make_cuda116cudnn840trt8406gcc82
;;
*
)
echo
"Make dockerfile error, Without this paramet."
exit
1
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录