提交 cecc4069 编写于 作者: M MRXLT

bug fix

上级 8e2c9b5d
...@@ -83,7 +83,7 @@ if (SERVER) ...@@ -83,7 +83,7 @@ if (SERVER)
OUTPUT ${PADDLE_SERVING_BINARY_DIR}/.timestamp OUTPUT ${PADDLE_SERVING_BINARY_DIR}/.timestamp
COMMAND cp -r COMMAND cp -r
${CMAKE_CURRENT_SOURCE_DIR}/paddle_serving_server_gpu/ ${PADDLE_SERVING_BINARY_DIR}/python/ ${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 COMMAND env ${py_env} ${PYTHON_EXECUTABLE} setup.py bdist_wheel
DEPENDS ${SERVING_SERVER_CORE} server_config_py_proto ${PY_FILES}) DEPENDS ${SERVING_SERVER_CORE} server_config_py_proto ${PY_FILES})
add_custom_target(paddle_python ALL DEPENDS ${PADDLE_SERVING_BINARY_DIR}/.timestamp) add_custom_target(paddle_python ALL DEPENDS ${PADDLE_SERVING_BINARY_DIR}/.timestamp)
......
...@@ -20,7 +20,7 @@ new_str = "" ...@@ -20,7 +20,7 @@ new_str = ""
with open("paddle_serving_server_gpu/version.py", "r") as f: with open("paddle_serving_server_gpu/version.py", "r") as f:
for line in f.readlines(): for line in f.readlines():
if re.match("cuda_version", line): 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 new_str = new_str + line
with open("paddle_serving_server_gpu/version.py", "w") as f: with open("paddle_serving_server_gpu/version.py", "w") as f:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册