Skip to content

Commit

Permalink
[CUDA][docs] explicitly require CUDA 9.0 or higher (#3885)
Browse files Browse the repository at this point in the history
  • Loading branch information
StrikerRUS authored Feb 1, 2021
1 parent e0887ae commit 763b5f3
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 5 deletions.
6 changes: 3 additions & 3 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -153,16 +153,16 @@ if(__INTEGRATE_OPENCL)
endif(__INTEGRATE_OPENCL)

if(USE_CUDA)
find_package(CUDA REQUIRED)
find_package(CUDA 9.0 REQUIRED)
include_directories(${CUDA_INCLUDE_DIRS})
LIST(APPEND CMAKE_CUDA_FLAGS -Xcompiler=${OpenMP_CXX_FLAGS} -Xcompiler=-fPIC -Xcompiler=-Wall)

set(CUDA_ARCHS "6.0" "6.1" "6.2" "7.0")
if(CUDA_VERSION VERSION_GREATER_EQUAL "10.0")
list(APPEND CUDA_ARCHS "7.5")
endif()
list(POP_BACK CUDA_ARCHS CUDA_LAST_SUPORTED_VERSION)
list(APPEND CUDA_ARCHS "${CUDA_LAST_SUPORTED_VERSION}+PTX")
list(POP_BACK CUDA_ARCHS CUDA_LAST_SUPPORTED_ARCH)
list(APPEND CUDA_ARCHS "${CUDA_LAST_SUPPORTED_ARCH}+PTX")
CUDA_SELECT_NVCC_ARCH_FLAGS(CUDA_ARCH_FLAGS ${CUDA_ARCHS})

LIST(APPEND CMAKE_CUDA_FLAGS ${CUDA_ARCH_FLAGS})
Expand Down
2 changes: 1 addition & 1 deletion docs/Installation-Guide.rst
Original file line number Diff line number Diff line change
Expand Up @@ -594,7 +594,7 @@ On Linux a CUDA version of LightGBM can be built using **CUDA**, **CMake** and *

The following dependencies should be installed before compilation:

- **CUDA** libraries. Please refer to `this detailed guide`_.
- **CUDA** 9.0 or later libraries. Please refer to `this detailed guide`_.

- **CMake** 3.16 or later.

Expand Down
2 changes: 1 addition & 1 deletion python-package/README.rst
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,7 @@ Build CUDA Version
All requirements from `Build from Sources section <#build-from-sources>`__ apply for this installation option as well, and `CMake`_ (version 3.16 or higher) is strongly required.

**CUDA** library is needed: details for installation can be found in `Installation Guide <https://github.com/microsoft/LightGBM/blob/master/docs/Installation-Guide.rst#build-cuda-version-experimental>`__.
**CUDA** library (version 9.0 or higher) is needed: details for installation can be found in `Installation Guide <https://github.com/microsoft/LightGBM/blob/master/docs/Installation-Guide.rst#build-cuda-version-experimental>`__.

Build HDFS Version
~~~~~~~~~~~~~~~~~~
Expand Down

0 comments on commit 763b5f3

Please sign in to comment.