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

modify paddle image version: cuda version->12.3

Browse Source
This commit is contained in:
xingjun.wxj
2025-04-09 11:23:09 +08:00
parent f2e3b4b664
commit ee5ecf46d8
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker/Dockerfile.ubuntu_paddle_gpu
View File

@@ -1,4 +1,4 @@
FROM paddlepaddle/paddle:3.0.0-gpu-cuda12.6-cudnn9.5-trt10.5
FROM paddlepaddle/paddle:3.0.0rc1-gpu-cuda12.3-cudnn9.0-trt8.6
ARG DEBIAN_FRONTEND=noninteractive
ENV TZ=Asia/Shanghai

4
docker/build_image.py
View File

@@ -458,8 +458,8 @@ class PaddleGPUImageBuilder(Builder):
self.args.python_tag = 'py310'
self.args.ubuntu_version = '20.04'
# self.args.modelscope_version = '1.24.0'
self.args.cuda_tag = '12.6'
self.args.trt_tag = '10.5'
self.args.cuda_tag = '12.3'
self.args.trt_tag = '8.6'
def generate_dockerfile(self) -> str:
with open('docker/Dockerfile.paddle_extra_install', 'r') as f: