| 214 | self.image(), '-f', 'Dockerfile', '.') |
| 215 | |
| 216 | def push(self): |
| 217 | ret = self.run_cmd('docker', 'push', self.image()) |
| 218 | if ret != 0: |
| 219 | return ret |
| 220 | image_tag2 = ( |
| 221 | f'{docker_registry}:ubuntu{self.args.ubuntu_version}-cuda{self.args.cuda_version}-' |
| 222 | f'{self.args.python_tag}-torch{self.args.torch_version}-tf{self.args.tf_version}-' |
| 223 | f'{self.args.modelscope_version}-{formatted_time}-test') |
| 224 | ret = self.run_cmd('docker', 'tag', self.image(), image_tag2) |
| 225 | if ret != 0: |
| 226 | return ret |
| 227 | return self.run_cmd('docker', 'push', image_tag2) |
| 228 | |
| 229 | |
| 230 | class BaseCPUImageBuilder(Builder): |