Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
Paddle
提交
ccf8d96c
P
Paddle
项目概览
BaiXuePrincess
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
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看板
未验证
提交
ccf8d96c
编写于
2月 01, 2023
作者:
P
pangyoki
提交者:
GitHub
2月 01, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
support cuda11.7 manylinux (#44217)
上级
3ab6faa8
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
18 addition
and
1 deletion
+18
-1
tools/dockerfile/build_scripts/install_cudnn.sh
tools/dockerfile/build_scripts/install_cudnn.sh
+8
-0
tools/dockerfile/build_scripts/install_nccl2.sh
tools/dockerfile/build_scripts/install_nccl2.sh
+1
-1
tools/dockerfile/centos7_manylinux.sh
tools/dockerfile/centos7_manylinux.sh
+9
-0
未找到文件。
tools/dockerfile/build_scripts/install_cudnn.sh
浏览文件 @
ccf8d96c
...
...
@@ -45,4 +45,12 @@ elif [[ "$1" == "cudnn821" && "$VERSION" == "11.2" ]]; then
cp
-r
lib64 /usr
&&
cd
../
&&
\
rm
-f
cudnn-11.3-linux-x64-v8.2.1.32.tgz
&&
\
rm
-rf
cuda
elif
[[
"
$1
"
==
"cudnn841"
&&
"
$VERSION
"
==
"11.7"
]]
;
then
wget
-q
https://paddle-ci.gz.bcebos.com/cudnn/cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz
--no-check-certificate
tar
xJvf cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz
&&
\
cd
cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive
&&
\
cp
-r
include /usr
&&
\
cp
-r
lib /usr
&&
cd
../
&&
\
rm
-f
cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz
&&
\
rm
-rf
cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive
fi
tools/dockerfile/build_scripts/install_nccl2.sh
浏览文件 @
ccf8d96c
...
...
@@ -17,7 +17,7 @@
VERSION
=
$(
nvcc
--version
|
grep
release |
grep
-oEi
"release ([0-9]+)
\.
([0-9])"
|
sed
"s/release //"
)
if
[
"
$VERSION
"
==
"10.0"
]
;
then
DEB
=
"nccl-repo-ubuntu1604-2.4.7-ga-cuda10.0_1-1_amd64.deb"
elif
[
"
$VERSION
"
==
"10.2"
]
||
[
"
$VERSION
"
==
"10.1"
]
||
[
"
$VERSION
"
==
"11.0"
]
||
[
"
$VERSION
"
==
"11.2"
]
||
[
"
$VERSION
"
==
"11.3"
]
||
[
"
$VERSION
"
==
"11.4"
]
||
[
"
$VERSION
"
==
"11.5"
]
||
[
"
$VERSION
"
==
"11.6"
]
||
[
"
$VERSION
"
==
"11.8"
]
;
then
elif
[
"
$VERSION
"
==
"10.2"
]
||
[
"
$VERSION
"
==
"10.1"
]
||
[
"
$VERSION
"
==
"11.0"
]
||
[
"
$VERSION
"
==
"11.2"
]
||
[
"
$VERSION
"
==
"11.3"
]
||
[
"
$VERSION
"
==
"11.4"
]
||
[
"
$VERSION
"
==
"11.5"
]
||
[
"
$VERSION
"
==
"11.6"
]
||
[
"
$VERSION
"
==
"11.
7"
]
||
[
"
$VERSION
"
==
"11.
8"
]
;
then
if
[
-f
"/etc/redhat-release"
]
;
then
rm
-f
/usr/local/lib/libnccl.so
wget
--no-check-certificate
-q
https://nccl2-deb.cdn.bcebos.com/libnccl-2.10.3-1+cuda11.4.x86_64.rpm
...
...
tools/dockerfile/centos7_manylinux.sh
浏览文件 @
ccf8d96c
...
...
@@ -128,6 +128,12 @@ function make_cuda116cudnn840trt8406gcc82() {
sed
-i
'/CMD/iRUN ldconfig'
Dockerfile.tmp
}
function
make_cuda117cudnn841
()
{
sed
's/<baseimg>/11.7.0-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/install_cudnn.sh cudnn841
\n
ENV CUDNN_VERSION=8.4.1
\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
...
...
@@ -188,6 +194,9 @@ function main() {
cuda116cudnn840trt8406gcc82
)
make_cuda116cudnn840trt8406gcc82
;;
cuda117cudnn841
)
make_cuda117cudnn841
;;
*
)
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.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录