Skip to content

Commit 8e94416

Browse files
authored
Publich torchserve nightly & release cpp dev docker images (#3023)
* Publich torchserve nightly cpp dev docker images * Create docker release images for cpp dev containers
1 parent c6c761b commit 8e94416

File tree

2 files changed

+42
-0
lines changed

2 files changed

+42
-0
lines changed

docker/build_upload_release.py

+20
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,14 @@
3636
f"./build_image.sh -g -cv cu121 -t {organization}/torchserve:latest-gpu",
3737
dry_run,
3838
)
39+
try_and_handle(
40+
f"./build_image.sh -bt dev -cpp -t {organization}/torchserve:latest-cpp-dev-cpu",
41+
dry_run,
42+
)
43+
try_and_handle(
44+
f"./build_image.sh -bt dev -g -cv cu121 -cpp -t {organization}/torchserve:latest-cpp-dev-gpu",
45+
dry_run,
46+
)
3947
try_and_handle(
4048
f"docker tag {organization}/torchserve:latest {organization}/torchserve:latest-cpu",
4149
dry_run,
@@ -48,13 +56,25 @@
4856
f"docker tag {organization}/torchserve:latest-gpu {organization}/torchserve:{check_ts_version()}-gpu",
4957
dry_run,
5058
)
59+
try_and_handle(
60+
f"docker tag {organization}/torchserve:latest-cpp-dev-cpu {organization}/torchserve:{check_ts_version()}-cpp-dev-cpu",
61+
dry_run,
62+
)
63+
try_and_handle(
64+
f"docker tag {organization}/torchserve:latest-cpp-dev-gpu {organization}/torchserve:{check_ts_version()}-cpp-dev-gpu",
65+
dry_run,
66+
)
5167

5268
for image in [
5369
f"{organization}/torchserve:latest",
5470
f"{organization}/torchserve:latest-cpu",
5571
f"{organization}/torchserve:latest-gpu",
72+
f"{organization}/torchserve:latest-cpp-dev-cpu",
73+
f"{organization}/torchserve:latest-cpp-dev-gpu",
5674
f"{organization}/torchserve:{check_ts_version()}-cpu",
5775
f"{organization}/torchserve:{check_ts_version()}-gpu",
76+
f"{organization}/torchserve:{check_ts_version()}-cpp-dev-cpu",
77+
f"{organization}/torchserve:{check_ts_version()}-cpp-dev-gpu",
5878
]:
5979
try_and_handle(f"docker push {image}", dry_run)
6080

docker/docker_nightly.py

+22
Original file line numberDiff line numberDiff line change
@@ -35,17 +35,29 @@
3535
project = "torchserve-nightly"
3636
cpu_version = f"{project}:cpu-{get_nightly_version()}"
3737
gpu_version = f"{project}:gpu-{get_nightly_version()}"
38+
cpp_dev_cpu_version = f"{project}:cpp-dev-cpu-{get_nightly_version()}"
39+
cpp_dev_gpu_version = f"{project}:cpp-dev-gpu-{get_nightly_version()}"
3840

3941
# Build Nightly images and append the date in the name
4042
try_and_handle(f"./build_image.sh -n -t {organization}/{cpu_version}", dry_run)
4143
try_and_handle(
4244
f"./build_image.sh -g -cv cu121 -n -t {organization}/{gpu_version}",
4345
dry_run,
4446
)
47+
try_and_handle(
48+
f"./build_image.sh -bt dev -cpp -t {organization}/{cpp_dev_cpu_version}",
49+
dry_run,
50+
)
51+
try_and_handle(
52+
f"./build_image.sh -bt dev -g -cv cu121 -cpp -t {organization}/{cpp_dev_gpu_version}",
53+
dry_run,
54+
)
4555

4656
# Push Nightly images to official PyTorch Dockerhub account
4757
try_and_handle(f"docker push {organization}/{cpu_version}", dry_run)
4858
try_and_handle(f"docker push {organization}/{gpu_version}", dry_run)
59+
try_and_handle(f"docker push {organization}/{cpp_dev_cpu_version}", dry_run)
60+
try_and_handle(f"docker push {organization}/{cpp_dev_gpu_version}", dry_run)
4961

5062
# Tag nightly images with latest
5163
try_and_handle(
@@ -56,10 +68,20 @@
5668
f"docker tag {organization}/{gpu_version} {organization}/{project}:latest-gpu",
5769
dry_run,
5870
)
71+
try_and_handle(
72+
f"docker tag {organization}/{cpp_dev_cpu_version} {organization}/{project}:latest-cpp-dev-cpu",
73+
dry_run,
74+
)
75+
try_and_handle(
76+
f"docker tag {organization}/{cpp_dev_gpu_version} {organization}/{project}:latest-cpp-dev-gpu",
77+
dry_run,
78+
)
5979

6080
# Push images with latest tag
6181
try_and_handle(f"docker push {organization}/{project}:latest-cpu", dry_run)
6282
try_and_handle(f"docker push {organization}/{project}:latest-gpu", dry_run)
83+
try_and_handle(f"docker push {organization}/{project}:latest-cpp-dev-cpu", dry_run)
84+
try_and_handle(f"docker push {organization}/{project}:latest-cpp-dev-gpu", dry_run)
6385

6486
# Cleanup built images
6587
if args.cleanup:

0 commit comments

Comments
 (0)