| 397 | 'Dockerfile', '.') |
| 398 | |
| 399 | def push(self): |
| 400 | ret = self.run_cmd('docker', 'push', self.image()) |
| 401 | if ret != 0: |
| 402 | return ret |
| 403 | image_tag2 = ( |
| 404 | f'{docker_registry}:ubuntu{self.args.ubuntu_version}-cuda{self.args.cuda_version}-' |
| 405 | f'{self.args.python_tag}-torch{self.args.torch_version}-' |
| 406 | f'{self.args.modelscope_version}-{formatted_time}-test') |
| 407 | ret = self.run_cmd('docker', 'tag', self.image(), image_tag2) |
| 408 | if ret != 0: |
| 409 | return ret |
| 410 | return self.run_cmd('docker', 'push', image_tag2) |
| 411 | |
| 412 | |
| 413 | class LatestGPUImageBuilder(StableGPUImageBuilder): |