bug fix

python/CMakeLists.txt

@@ -83,7 +83,7 @@ if (SERVER)
             OUTPUT ${PADDLE_SERVING_BINARY_DIR}/.timestamp
             COMMAND cp -r
             ${CMAKE_CURRENT_SOURCE_DIR}/paddle_serving_server_gpu/ ${PADDLE_SERVING_BINARY_DIR}/python/
-            COMMAND env ${py_env} ${PYTHON_EXECUTABLE} paddle_serving_server_gpu/gen_cuda_version.py
+            COMMAND env ${py_env} ${PYTHON_EXECUTABLE} paddle_serving_server_gpu/gen_cuda_version.py ${CUDA_VERSION_MAJOR}
             COMMAND env ${py_env} ${PYTHON_EXECUTABLE} setup.py bdist_wheel
             DEPENDS ${SERVING_SERVER_CORE} server_config_py_proto ${PY_FILES})
         add_custom_target(paddle_python ALL DEPENDS ${PADDLE_SERVING_BINARY_DIR}/.timestamp)

python/paddle_serving_server_gpu/gen_cuda_version.py

@@ -20,7 +20,7 @@ new_str = ""
 with open("paddle_serving_server_gpu/version.py", "r") as f:
     for line in f.readlines():
         if re.match("cuda_version", line):
-            line = re.sub(r"\d+", "10", line)
+            line = re.sub(r"\d+", sys.argv[1], line)
         new_str = new_str + line
 
 with open("paddle_serving_server_gpu/version.py", "w") as f: