Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 04:01:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

recover py tag for base gpu image

Browse Source
This commit is contained in:
xingjun.wxj
2025-03-11 17:06:51 +08:00
parent a687a212ff
commit 7debaac31b
1 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docker/build_image.py
View File

@@ -129,13 +129,8 @@ 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}-py310-'
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')
def build(self) -> int: