python package name gpu

8 years ago · 361126f26b
parent 01e28d1125
commit 361126f26b
2 changed files with 7 additions and 1 deletions
--- a/python/CMakeLists.txt
+++ b/python/CMakeLists.txt
@ -33,6 +33,12 @@ if(WITH_MKLDNN)
  list(APPEND MKL_DEPENDS mkldnn)
 endif()

+if(WITH_GPU)
+  SET(PACKAGE_NAME "paddlepaddle_gpu")
+else()
+  SET(PACKAGE_NAME "paddlepaddle")
+endif()
+
 configure_file(${CMAKE_CURRENT_SOURCE_DIR}/setup.py.in
    ${CMAKE_CURRENT_BINARY_DIR}/setup.py)

--- a/python/setup.py.in
+++ b/python/setup.py.in
@ -89,7 +89,7 @@ paddle_rt_libs = ['${WARPCTC_LIBRARIES}']
 if '${MKL_SHARED_LIBS}'!= '':
  paddle_rt_libs += '${MKL_SHARED_LIBS}'.split(';')

-setup(name='paddlepaddle',
+setup(name='${PACKAGE_NAME}',
      version='${PADDLE_VERSION}',
      description='Parallel Distributed Deep Learning',
      install_requires=setup_requires,