Skip to content

Commit 29c20f6

Browse files
XieYunshenXieYunshen
authored
编译镜像切换base镜像Almalinux,新增cuda12.4,12.6和12.8镜像 (#3032)
1 parent 39e76af commit 29c20f6

File tree

1 file changed

+41
-20
lines changed

1 file changed

+41
-20
lines changed

‎tools/linux_env_info.sh

+41-20
Original file line numberDiff line numberDiff line change
@@ -29,56 +29,77 @@ function DockerImages () {
2929

3030
if [[ "${docker_type}" == "Manual" ]];then
3131
echo "Select Manual Mode!!!"
32-
elif [[ "${docker_type}" == "Centos" ]];then
32+
elif [[ "${docker_type}" == "Centos" || "${docker_type}" == "Almalinux" ]];then
3333
case ${cuda_version} in
3434
"Cuda102")
3535
echo "Selected Centos: Cuda102"
36-
export Image_version="registry.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda10.2-cudnn7.6-trt7.0-gcc8.2"
36+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda10.2-cudnn7.6-trt7.0-gcc8.2"
3737
export env_cuda_version="10.2"
3838
export env_cudnn_version="7.6.5"
3939
export env_trt_version="7.0.0.11"
4040
;;
4141
"Cuda112")
4242
echo "Selected Centos: Cuda112"
43-
export Image_version="registry.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda11.2-cudnn8.2-trt8.0-gcc82"
43+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda11.2-cudnn8.2-trt8.0-gcc82"
4444
export env_cuda_version="11.2"
4545
export env_cudnn_version="8.2.1"
4646
export env_trt_version="8.0.3.4"
4747
;;
4848
"Cuda116")
4949
echo "Selected Centos: Cuda116"
50-
export Image_version="registry.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda11.6-cudnn8.4-trt8.4-gcc8.2"
50+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda11.6-cudnn8.4-trt8.4-gcc8.2"
5151
export env_cuda_version="11.6"
5252
export env_cudnn_version="8.4.0"
5353
export env_trt_version="8.4.0.6"
5454
;;
5555
"Cuda117")
5656
echo "Selected Centos: Cuda117"
57-
export Image_version="registry.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda11.7-cudnn8.4-trt8.4-gcc8.2"
57+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda11.7-cudnn8.4-trt8.4-gcc8.2"
5858
export env_cuda_version="11.7"
5959
export env_cudnn_version="8.4.1"
6060
export env_trt_version="8.4.2.4"
6161
;;
6262
"Cuda118")
63-
echo "Selected Centos: Cuda118"
64-
export Image_version="registry.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda11.8-cudnn8.6-trt8.5-gcc8.2"
63+
echo "Selected Almalinux: Cuda118"
64+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda11.8-cudnn8.9-trt8.6-gcc11"
6565
export env_cuda_version="11.8"
66-
export env_cudnn_version="8.6.0"
67-
export env_trt_version="8.5.3.1"
66+
export env_cudnn_version="8.9.7"
67+
export env_trt_version="8.6.1.6"
6868
;;
6969
"Cuda120")
7070
echo "Selected Centos: Cuda120"
71-
export Image_version="registry.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda12.0-cudnn8.9-trt8.6-gcc12.2"
71+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda12.0-cudnn8.9-trt8.6-gcc12.2"
7272
export env_cuda_version="12.0"
7373
export env_cudnn_version="8.9.1"
7474
export env_trt_version="8.6.1.6"
7575
;;
7676
"Cuda123")
7777
echo "Selected Centos: Cuda123"
78-
export Image_version="registry.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda12.3-cudnn9.0-trt8.6-gcc12.2"
78+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda12.3-cudnn9.1-trt10.5-gcc11"
7979
export env_cuda_version="12.3"
80-
export env_cudnn_version="9.0.0"
81-
export env_trt_version="8.6.1.6"
80+
export env_cudnn_version="9.1.0"
81+
export env_trt_version="10.5.0.18"
82+
;;
83+
"Cuda124")
84+
echo "Selected Almalinux: Cuda124"
85+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda12.4-cudnn9.1-trt10.5-gcc11"
86+
export env_cuda_version="12.4"
87+
export env_cudnn_version="9.1.1"
88+
export env_trt_version="10.5.0.18"
89+
;;
90+
"Cuda126")
91+
echo "Selected Almalinux: Cuda126"
92+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda12.6-cudnn9.5-trt10.5-gcc11"
93+
export env_cuda_version="12.6"
94+
export env_cudnn_version="9.5.1"
95+
export env_trt_version="10.5.0.18"
96+
;;
97+
"Cuda128")
98+
echo "Selected Almalinux: Cuda128"
99+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle_manylinux_devel:cuda12.8-cudnn9.7-trt10.5-gcc11"
100+
export env_cuda_version="12.8"
101+
export env_cudnn_version="9.7.0"
102+
export env_trt_version="10.5.0.18"
82103
;;
83104
*)
84105
DOCKER_EXIT_CODE=101
@@ -89,49 +110,49 @@ function DockerImages () {
89110
case ${cuda_version} in
90111
"Cuda102")
91112
echo "Selected Ubuntu: Cuda102"
92-
export Image_version="registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda10.2-cudnn7.6-trt7.0-gcc8.2"
113+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:latest-dev-cuda10.2-cudnn7.6-trt7.0-gcc8.2"
93114
export env_cuda_version="10.2"
94115
export env_cudnn_version="7.6.5"
95116
export env_trt_version="7.0.0.11"
96117
;;
97118
"Cuda112")
98119
echo "Selected Ubuntu: Cuda112"
99-
export Image_version="registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.2-cudnn8.2-trt8.0-gcc82"
120+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.2-cudnn8.2-trt8.0-gcc82"
100121
export env_cuda_version="11.2"
101122
export env_cudnn_version="8.2.1"
102123
export env_trt_version="8.0.3.4"
103124
;;
104125
"Cuda116")
105126
echo "Selected Ubuntu: Cuda116"
106-
export Image_version="registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.6-cudnn8.4-trt8.4-gcc82"
127+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.6-cudnn8.4-trt8.4-gcc82"
107128
export env_cuda_version="11.6"
108129
export env_cudnn_version="8.4.0"
109130
export env_trt_version="8.4.0.6"
110131
;;
111132
"Cuda117")
112133
echo "Selected Ubuntu: Cuda117"
113-
export Image_version="registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.7-cudnn8.4-trt8.4-gcc82"
134+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.7-cudnn8.4-trt8.4-gcc82"
114135
export env_cuda_version="11.7"
115136
export env_cudnn_version="8.4.1"
116137
export env_trt_version="8.4.2.4"
117138
;;
118139
"Cuda118")
119140
echo "Selected Ubuntu: Cuda118"
120-
export Image_version="registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.8-cudnn8.6-trt8.5-gcc82"
141+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.8-cudnn8.6-trt8.5-gcc82"
121142
export env_cuda_version="11.8"
122143
export env_cudnn_version="8.6.0"
123144
export env_trt_version="8.5.3.1"
124145
;;
125146
"Cuda120")
126147
echo "Selected Ubuntu: Cuda120"
127-
export Image_version="registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda12.0-cudnn8.9-trt8.6-gcc12.2"
148+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:latest-dev-cuda12.0-cudnn8.9-trt8.6-gcc12.2"
128149
export env_cuda_version="12.0"
129150
export env_cudnn_version="8.9.1"
130151
export env_trt_version="8.6.1.6"
131152
;;
132153
"Cuda123")
133154
echo "Selected Ubuntu: Cuda123"
134-
export Image_version="registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda12.3-cudnn9.0-trt8.6-gcc12.2"
155+
export Image_version="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:latest-dev-cuda12.3-cudnn9.0-trt8.6-gcc12.2"
135156
export env_cuda_version="12.3"
136157
export env_cudnn_version="9.0.0"
137158
export env_trt_version="8.6.1.6"

0 commit comments

Comments
 (0)