Skip to content

Commit 2152a18

Browse files
authored
Update python to 3.13 and cuda to 12.8 (#126)
* upgrade python and torch * upgrade py and torch * vllm only supports py 3.12 * update README
1 parent 588a32b commit 2152a18

File tree

2 files changed

+25
-90
lines changed

2 files changed

+25
-90
lines changed

.github/workflows/build-docker-gpu.yml

Lines changed: 22 additions & 89 deletions
Original file line numberDiff line numberDiff line change
@@ -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

README.md

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,4 +27,6 @@ LabNow —— 一键开启的数据科学与 AI 环境,加速你的智能创
2727

2828
## Documentation & Tutorial
2929

30-
[Wiki & Document](https://LabNow.ai) | [中文使用指引(含中国网络镜像)](https://labnow-ai.feishu.cn/wiki/wikcn0sBhMtb1KNRSUTettxWstc)
30+
[Wiki & Document](https://LabNow.ai)
31+
| [中文使用指引(含中国网络镜像)](https://labnow-ai.feishu.cn/wiki/wikcn0sBhMtb1KNRSUTettxWstc)
32+
| [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/LabNow-ai/lab-foundation)

0 commit comments

Comments
 (0)