提交 361126f2 编写于 作者: T typhoonzero

python package name gpu

上级 01e28d11
...@@ -33,6 +33,12 @@ if(WITH_MKLDNN) ...@@ -33,6 +33,12 @@ if(WITH_MKLDNN)
list(APPEND MKL_DEPENDS mkldnn) list(APPEND MKL_DEPENDS mkldnn)
endif() endif()
if(WITH_GPU)
SET(PACKAGE_NAME "paddlepaddle_gpu")
else()
SET(PACKAGE_NAME "paddlepaddle")
endif()
configure_file(${CMAKE_CURRENT_SOURCE_DIR}/setup.py.in configure_file(${CMAKE_CURRENT_SOURCE_DIR}/setup.py.in
${CMAKE_CURRENT_BINARY_DIR}/setup.py) ${CMAKE_CURRENT_BINARY_DIR}/setup.py)
......
...@@ -89,7 +89,7 @@ paddle_rt_libs = ['${WARPCTC_LIBRARIES}'] ...@@ -89,7 +89,7 @@ paddle_rt_libs = ['${WARPCTC_LIBRARIES}']
if '${MKL_SHARED_LIBS}'!= '': if '${MKL_SHARED_LIBS}'!= '':
paddle_rt_libs += '${MKL_SHARED_LIBS}'.split(';') paddle_rt_libs += '${MKL_SHARED_LIBS}'.split(';')
setup(name='paddlepaddle', setup(name='${PACKAGE_NAME}',
version='${PADDLE_VERSION}', version='${PADDLE_VERSION}',
description='Parallel Distributed Deep Learning', description='Parallel Distributed Deep Learning',
install_requires=setup_requires, install_requires=setup_requires,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册