From a687a212ff2d04d971103bbbfd0d41e3f3b45d7f Mon Sep 17 00:00:00 2001 From: "xingjun.wxj" Date: Tue, 11 Mar 2025 17:01:04 +0800 Subject: [PATCH] set py310 for gpu base image --- docker/build_image.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/docker/build_image.py b/docker/build_image.py index 5f253eae..231eb685 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -129,8 +129,13 @@ class BaseGPUImageBuilder(Builder): return content def image(self) -> str: + # return ( + # f'{docker_registry}:ubuntu{self.args.ubuntu_version}-cuda{self.args.cuda_version}-{self.args.python_tag}-' + # f'torch{self.args.torch_version}-tf{self.args.tf_version}-base') + + # TODO: set the python tag to `py310` for base image temporarily return ( - f'{docker_registry}:ubuntu{self.args.ubuntu_version}-cuda{self.args.cuda_version}-{self.args.python_tag}-' + f'{docker_registry}:ubuntu{self.args.ubuntu_version}-cuda{self.args.cuda_version}-py310-' f'torch{self.args.torch_version}-tf{self.args.tf_version}-base') def build(self) -> int: