modify suffix for v0.7

python/CMakeLists.txt

@@ -72,9 +72,13 @@ if (SERVER)
     if(CUDA_VERSION EQUAL 10.1)
       set(VERSION_SUFFIX 101)
     elseif(CUDA_VERSION EQUAL 10.2)
-      set(VERSION_SUFFIX 102)
-    elseif(CUDA_VERSION EQUAL 11.0)
-      set(VERSION_SUFFIX 11)
+      if(CUDNN_MAJOR_VERSION EQUAL 7)
+        set(VERSION_SUFFIX 1027)
+      elif(CUDNN_MAJOR_VERSION EQUAL 8)
+        set(VERSION_SUFFIX 1028)
+      endif()
+    elseif(CUDA_VERSION EQUAL 11.2)
+      set(VERSION_SUFFIX 112)
     endif()
   endif()
 

python/paddle_serving_server/server.py

@@ -428,7 +428,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 == "102":
+            if version_suffix == "101" or version_suffix == "1027" or version_suffix == "1028" or version_suffix == "112":
                 device_version = "gpu-" + version_suffix
             else:
                 device_version = "gpu-cuda" + version_suffix