diff --git a/python/CMakeLists.txt b/python/CMakeLists.txt index 86740ce6033f9dbab490e9b9df21ffd17ebe0cd0..986fd8ee289677234407d7ee2fc4d492933731ba 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 909cb8764bb7572af079e9a51c4f76dbd86441e0..c6446ac2dc61fa7a2123e6713b84624776435863 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