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

set py310 for gpu base image

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

7
docker/build_image.py
View File

@@ -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: