@@ -33,23 +33,10 @@ jobs:
3333 # latest cuda supported by tensorflow: https://tensorflow.google.cn/install/source?hl=en#gpu
3434 # latest cuda supported by paddlepadle: https://www.paddlepaddle.org.cn/
3535 # latest cuda supported by vllm: https://docs.vllm.ai/en/latest/getting_started/installation/gpu.html?device=cuda
36- job-cuda_126 :
37- name : ' cuda_12.6,cuda,nvidia-cuda'
38- runs-on : ubuntu-latest
39- steps :
40- - uses : actions/checkout@v4
41- - run : |
42- source ./tool.sh
43- build_image_no_tag nvidia-cuda 12.6.3-cudnn-devel-ubuntu24.04 docker_atom/Dockerfile --build-arg "BASE_IMG=nvidia/cuda:12.6.3-cudnn-devel-ubuntu24.04" && clear_images nvidia/cuda
44- export IMG_PREFIX_SRC="${IMG_PREFIX_DST}"
45- build_image_no_tag tmp latest docker_base/Dockerfile --build-arg "BASE_IMG=nvidia-cuda:12.6.3-cudnn-devel-ubuntu24.04"
46- build_image cuda_12.6 latest docker_cuda/nvidia-cuda.Dockerfile --build-arg "BASE_IMG=tmp"
47- alias_image cuda_12.6 latest cuda latest
48- push_image cuda
4936
5037 # reserved for vllm: https://github.com/vllm-project/vllm/blob/main/docker/Dockerfile
5138 job-cuda_128 :
52- name : ' cuda_12.8'
39+ name : ' cuda_12.8,cuda,nvidia-cuda '
5340 runs-on : ubuntu-latest
5441 steps :
5542 - uses : actions/checkout@v4
@@ -61,50 +48,20 @@ jobs:
6148 build_image cuda_12.8 latest docker_cuda/nvidia-cuda.Dockerfile --build-arg "BASE_IMG=tmp"
6249 push_image cuda
6350
64- # reserved for paddlepaddl 2.6: https://www.paddlepaddle.org.cn
65- job-cuda_120 :
66- name : ' cuda_12.0'
67- runs-on : ubuntu-latest
68- steps :
69- - uses : actions/checkout@v4
70- - run : |
71- source ./tool.sh
72- build_image_no_tag nvidia-cuda 12.0.1-cudnn8-devel-ubuntu22.04 docker_atom/Dockerfile --build-arg "BASE_IMG=nvidia/cuda:12.0.1-cudnn8-devel-ubuntu22.04" && clear_images nvidia/cuda
73- export IMG_PREFIX_SRC="${IMG_PREFIX_DST}"
74- build_image_no_tag tmp latest docker_base/Dockerfile --build-arg "BASE_IMG=nvidia-cuda:12.0.1-cudnn8-devel-ubuntu22.04"
75- build_image cuda_12.0 latest docker_cuda/nvidia-cuda.Dockerfile --build-arg "BASE_IMG=tmp"
76- push_image cuda
77-
78- # reserved for paddlepaddl 2.6, torch, and vllm
79- job-cuda_118 :
80- name : ' cuda_11.8'
81- runs-on : ubuntu-latest
82- steps :
83- - uses : actions/checkout@v4
84- - run : |
85- source ./tool.sh
86- build_image_no_tag nvidia-cuda 11.8.0-cudnn8-devel-ubuntu22.04 docker_atom/Dockerfile --build-arg "BASE_IMG=nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04" && clear_images nvidia/cuda
87- export IMG_PREFIX_SRC="${IMG_PREFIX_DST}"
88- build_image_no_tag tmp latest docker_base/Dockerfile --build-arg "BASE_IMG=nvidia-cuda:11.8.0-cudnn8-devel-ubuntu22.04"
89- build_image cuda_11.8 latest docker_cuda/nvidia-cuda.Dockerfile --build-arg "BASE_IMG=tmp"
90- push_image cuda
91-
92-
93- # reserved for tensorflow 1.x
94- job-cuda_112 :
95- name : ' cuda_11.2'
51+ job-cuda_126 :
52+ name : ' cuda_12.6'
9653 runs-on : ubuntu-latest
9754 steps :
9855 - uses : actions/checkout@v4
9956 - run : |
10057 source ./tool.sh
101- build_image_no_tag nvidia-cuda 11.2.2-cudnn8 -devel-ubuntu20 .04 docker_atom/Dockerfile --build-arg "BASE_IMG=nvidia/cuda:11.2.2-cudnn8 -devel-ubuntu20 .04" && clear_images nvidia/cuda
58+ build_image_no_tag nvidia-cuda 12.6.3-cudnn -devel-ubuntu24 .04 docker_atom/Dockerfile --build-arg "BASE_IMG=nvidia/cuda:12.6.3-cudnn -devel-ubuntu24 .04" && clear_images nvidia/cuda
10259 export IMG_PREFIX_SRC="${IMG_PREFIX_DST}"
103- build_image_no_tag tmp latest docker_base/Dockerfile --build-arg "BASE_IMG=nvidia-cuda:11.2.2-cudnn8-devel-ubuntu20.04" --build-arg "PYTHON_VERSION=3.8"
104- build_image cuda_11.2 latest docker_cuda/nvidia-cuda.Dockerfile --build-arg "BASE_IMG=tmp"
60+ build_image_no_tag tmp latest docker_base/Dockerfile --build-arg "BASE_IMG=nvidia-cuda:12.6.3-cudnn-devel-ubuntu24.04"
61+ build_image cuda_12.6 latest docker_cuda/nvidia-cuda.Dockerfile --build-arg "BASE_IMG=tmp"
62+ alias_image cuda_12.6 latest cuda latest
10563 push_image cuda
10664
107-
10865 job-tf2 :
10966 name : ' tf2,tf2-cuda126'
11067 needs : job-cuda_126
@@ -117,28 +74,16 @@ jobs:
11774 alias_image tf2-cuda126 latest tf2 latest
11875 push_image
11976
120- job-torch_cuda126 :
121- name : ' torch,torch-cuda126'
122- needs : job-cuda_126
123- runs-on : ubuntu-latest
124- steps :
125- - uses : actions/checkout@v4
126- - run : |
127- source ./tool.sh
128- build_image torch-cuda126 latest docker_core/Dockerfile --build-arg "BASE_IMG=cuda_12.6" --build-arg "ARG_PROFILE_PYTHON=torch"
129- alias_image torch-cuda126 latest torch latest
130- push_image
131-
132- job-paddle_cuda120 :
133- name : ' paddle-cuda120,paddle-2.6'
134- needs : job-cuda_120
77+ job-torch_cuda128 :
78+ name : ' torch,torch-cuda128'
79+ needs : job-cuda_128
13580 runs-on : ubuntu-latest
13681 steps :
13782 - uses : actions/checkout@v4
13883 - run : |
13984 source ./tool.sh
140- build_image paddle-cuda120 latest docker_core/Dockerfile --build-arg "BASE_IMG=cuda_12.0 " --build-arg "ARG_PROFILE_PYTHON=paddle,mkl "
141- alias_image paddle-cuda120 latest paddle-2.6 latest
85+ build_image torch-cuda128 latest docker_core/Dockerfile --build-arg "BASE_IMG=cuda_12.8 " --build-arg "ARG_PROFILE_PYTHON=torch "
86+ alias_image torch-cuda128 latest torch latest
14287 push_image
14388
14489 job-paddle_cuda126 :
@@ -155,57 +100,45 @@ jobs:
155100
156101
157102 job-py-nlp :
158- name : ' py-nlp,py-nlp-cuda126'
159- needs : job-cuda_126
160- runs-on : ubuntu-latest
161- steps :
162- - uses : actions/checkout@v4
163- - run : |
164- source ./tool.sh && export IMG_PREFIX_SRC="${IMG_PREFIX_DST}"
165- build_image py-nlp-cuda126 latest docker_core/Dockerfile --build-arg "BASE_IMG=cuda_12.6" --build-arg "ARG_PROFILE_PYTHON=datascience,mkl,torch,nlp"
166- alias_image py-nlp-cuda126 latest py-nlp latest
167- push_image
168-
169- job-py-nlp-cuda128 :
170- name : ' py-nlp-cuda128'
103+ name : ' py-nlp,py-nlp-cuda128'
171104 needs : job-cuda_128
172105 runs-on : ubuntu-latest
173106 steps :
174107 - uses : actions/checkout@v4
175108 - run : |
176109 source ./tool.sh && export IMG_PREFIX_SRC="${IMG_PREFIX_DST}"
177110 build_image py-nlp-cuda128 latest docker_core/Dockerfile --build-arg "BASE_IMG=cuda_12.8" --build-arg "ARG_PROFILE_PYTHON=datascience,mkl,torch,nlp"
111+ alias_image py-nlp-cuda128 latest py-nlp latest
178112 push_image
179113
180-
181114 job-py-cv :
182115 name : ' py-cv'
183- needs : job-cuda_126
116+ needs : job-cuda_128
184117 runs-on : ubuntu-latest
185118 steps :
186119 - uses : actions/checkout@v4
187120 - run : |
188121 source ./tool.sh && export IMG_PREFIX_SRC="${IMG_PREFIX_DST}"
189- build_image py-cv latest docker_core/Dockerfile --build-arg "BASE_IMG=cuda_12.6 " --build-arg "ARG_PROFILE_PYTHON=datascience,mkl,torch,cv"
122+ build_image py-cv latest docker_core/Dockerfile --build-arg "BASE_IMG=cuda_12.8 " --build-arg "ARG_PROFILE_PYTHON=datascience,mkl,torch,cv"
190123 push_image
191124
192125
193126 job-core-cuda :
194- name : ' core-cuda,full-cuda-12.6 '
195- needs : job-cuda_126
127+ name : ' core-cuda,full-cuda-12.8 '
128+ needs : job-cuda_128
196129 runs-on : ubuntu-latest
197130 steps :
198131 - uses : actions/checkout@v4
199132 - run : |
200133 source ./tool.sh && free_diskspace && export IMG_PREFIX_SRC="${IMG_PREFIX_DST}"
201- build_image full-cuda-12.6 latest docker_core/Dockerfile \
202- --build-arg "BASE_IMG=cuda_12.6 " \
134+ build_image full-cuda-12.8 latest docker_core/Dockerfile \
135+ --build-arg "BASE_IMG=cuda_12.8 " \
203136 --build-arg "ARG_PROFILE_PYTHON=base,datascience,mkl,database,nlp,cv,chem,tf2,torch" \
204137 --build-arg "ARG_PROFILE_R=base,datascience" \
205138 --build-arg "ARG_PROFILE_NODEJS=base" \
206139 --build-arg "ARG_PROFILE_JAVA=base,maven" \
207140 --build-arg "ARG_PROFILE_LATEX=base,cjk"
208- alias_image full-cuda-12.6 latest core-cuda latest && push_image cuda
141+ alias_image full-cuda-12.8 latest core-cuda latest && push_image cuda
209142
210143 job-nvidia-ctk :
211144 name : ' nvidia-ctk'
@@ -219,7 +152,7 @@ jobs:
219152
220153 # # Sync all images in this build (listed by "names") to mirror registry.
221154 sync_images :
222- needs : ["job-core-cuda", "job-py-cv", "job-py-nlp", "job-torch_cuda126 ", "job-nvidia-ctk"]
155+ needs : ["job-core-cuda", "job-py-cv", "job-py-nlp", "job-torch_cuda128 ", "job-nvidia-ctk"]
223156 runs-on : ubuntu-latest
224157 steps :
225158 - uses : actions/checkout@v4
0 commit comments