| 157 | self.image(), '-f', 'Dockerfile', '.') |
| 158 | |
| 159 | def push(self): |
| 160 | ret = self.run_cmd('docker', 'push', self.image()) |
| 161 | if ret != 0: |
| 162 | return ret |
| 163 | image_tag2 = ( |
| 164 | f'{docker_registry}:ubuntu{self.args.ubuntu_version}-{self.args.python_tag}-' |
| 165 | f'torch{self.args.torch_version}-{self.args.modelscope_version}-{formatted_time}-test' |
| 166 | ) |
| 167 | ret = self.run_cmd('docker', 'tag', self.image(), image_tag2) |
| 168 | if ret != 0: |
| 169 | return ret |
| 170 | return self.run_cmd('docker', 'push', image_tag2) |
| 171 | |
| 172 | |
| 173 | class OldGPUImageBuilder(Builder): |