Skip to content

Commit b478968

Browse files
committed
Publich torchserve nightly cpp dev docker images
1 parent 3694b11 commit b478968

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

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)