From 41e9bc5e0e871b4722243af4b2b3f37b9bf3d6b8 Mon Sep 17 00:00:00 2001 From: bjjwwang Date: Mon, 15 Nov 2021 21:13:32 +0800 Subject: [PATCH] rename 1027 to 102 --- python/CMakeLists.txt | 2 +- python/paddle_serving_server/server.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/CMakeLists.txt b/python/CMakeLists.txt index 86740ce6..986fd8ee 100644 --- a/python/CMakeLists.txt +++ b/python/CMakeLists.txt @@ -73,7 +73,7 @@ if (SERVER) set(VERSION_SUFFIX 101) elseif(CUDA_VERSION EQUAL 10.2) if(CUDNN_MAJOR_VERSION EQUAL 7) - set(VERSION_SUFFIX 1027) + set(VERSION_SUFFIX 102) elseif(CUDNN_MAJOR_VERSION EQUAL 8) set(VERSION_SUFFIX 1028) endif() diff --git a/python/paddle_serving_server/server.py b/python/paddle_serving_server/server.py index 909cb876..c6446ac2 100755 --- a/python/paddle_serving_server/server.py +++ b/python/paddle_serving_server/server.py @@ -429,7 +429,7 @@ class Server(object): if device_type == "0": device_version = self.get_device_version() elif device_type == "1": - if version_suffix == "101" or version_suffix == "1027" or version_suffix == "1028" or version_suffix == "112": + if version_suffix == "101" or version_suffix == "102" or version_suffix == "1028" or version_suffix == "112": device_version = "gpu-" + version_suffix else: device_version = "gpu-cuda" + version_suffix -- GitLab