未验证 提交 ccf8d96c 编写于 作者: P pangyoki 提交者: GitHub

support cuda11.7 manylinux (#44217)

上级 3ab6faa8
......@@ -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
......@@ -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
......
......@@ -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 \nRUN mv /usr/bin/cc /usr/bin/cc.bak \&\& ln -s /usr/local/gcc-8.2/bin/gcc /usr/bin/cc \nENV PATH=/usr/local/gcc-8.2/bin:\$PATH \nRun bash build_scripts/install_cudnn.sh cudnn841 \nENV CUDNN_VERSION=8.4.1 \nRUN 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.
先完成此消息的编辑!
想要评论请 注册