|
36 | 36 | f"./build_image.sh -g -cv cu121 -t {organization}/torchserve:latest-gpu",
|
37 | 37 | dry_run,
|
38 | 38 | )
|
| 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 | + ) |
39 | 47 | try_and_handle(
|
40 | 48 | f"docker tag {organization}/torchserve:latest {organization}/torchserve:latest-cpu",
|
41 | 49 | dry_run,
|
|
48 | 56 | f"docker tag {organization}/torchserve:latest-gpu {organization}/torchserve:{check_ts_version()}-gpu",
|
49 | 57 | dry_run,
|
50 | 58 | )
|
| 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 | + ) |
51 | 67 |
|
52 | 68 | for image in [
|
53 | 69 | f"{organization}/torchserve:latest",
|
54 | 70 | f"{organization}/torchserve:latest-cpu",
|
55 | 71 | f"{organization}/torchserve:latest-gpu",
|
| 72 | + f"{organization}/torchserve:latest-cpp-dev-cpu", |
| 73 | + f"{organization}/torchserve:latest-cpp-dev-gpu", |
56 | 74 | f"{organization}/torchserve:{check_ts_version()}-cpu",
|
57 | 75 | 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", |
58 | 78 | ]:
|
59 | 79 | try_and_handle(f"docker push {image}", dry_run)
|
60 | 80 |
|
|
0 commit comments