diff --git a/docker/Dockerfile.ubuntu_paddle_gpu b/docker/Dockerfile.ubuntu_paddle_gpu index 1c30c936..33f8761d 100644 --- a/docker/Dockerfile.ubuntu_paddle_gpu +++ b/docker/Dockerfile.ubuntu_paddle_gpu @@ -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 diff --git a/docker/build_image.py b/docker/build_image.py index 3f0ffc99..8a2c0917 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -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: