Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-05-18 21:25:26 +02:00
Code Issues Packages Projects Releases Wiki Activity

update cuda13.0

Browse Source
This commit is contained in:
Jintao Huang
2026-04-30 23:16:26 +08:00
parent 5c160e97c4
commit 537722acea
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docker/build_image.py
View File

@@ -401,9 +401,9 @@ class LatestGPUImageBuilder(StableGPUImageBuilder):
args.vllm_version = '0.19.1'
if not args.base_image:
# A mirrored image of nvidia/cuda:12.4.0-devel-ubuntu22.04
args.base_image = 'nvidia/cuda:12.9.1-cudnn-devel-ubuntu22.04'
args.base_image = 'nvidia/cuda:13.0.3-cudnn-devel-ubuntu22.04'
if not args.cuda_version:
args.cuda_version = '12.9.1'
args.cuda_version = '13.0.3'
return super().init_args(args)
def generate_dockerfile(self) -> str: