From ee81605e5d38c9a44a1d7601433641d4aa84527d Mon Sep 17 00:00:00 2001 From: WJJ1995 Date: Tue, 23 May 2023 15:16:07 +0800 Subject: [PATCH] [Serving]Support 11.2 CUDA (#1976) * add workspace_size option * support 11.2 cuda --------- Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com> --- serving/Dockerfile | 2 +- serving/scripts/build.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) mode change 100644 => 100755 serving/Dockerfile mode change 100644 => 100755 serving/scripts/build.sh diff --git a/serving/Dockerfile b/serving/Dockerfile old mode 100644 new mode 100755 index c2a7f5677..110f68a7e --- a/serving/Dockerfile +++ b/serving/Dockerfile @@ -13,7 +13,7 @@ # limitations under the License. -FROM nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04 +FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04 ARG http_proxy ARG https_proxy diff --git a/serving/scripts/build.sh b/serving/scripts/build.sh old mode 100644 new mode 100755 index ffcd43e36..9e7fdecbe --- a/serving/scripts/build.sh +++ b/serving/scripts/build.sh @@ -86,7 +86,7 @@ nvidia-docker run -i --rm --name ${docker_name} \ -e "http_proxy=${http_proxy}" \ -e "https_proxy=${https_proxy}" \ -e "trt_version=${trt_version}"\ - nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04 \ + nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04 \ bash -c \ 'export https_proxy_tmp=${https_proxy} export http_proxy_tmp=${http_proxy}