Support building with CTK11.5. (#7379)

* Support building with CTK11.5. * Require system cub installation for CTK11.4+. * Check thrust version for segmented sort.
2021-11-02 16:22:26 +08:00
parent a13321148a
commit 32e673d8c4
3 changed files with 35 additions and 11 deletions
--- a/doc/build.rst
+++ b/doc/build.rst
@@ -148,7 +148,8 @@ From the command line on Linux starting from the XGBoost directory:

  mkdir build
  cd build
-  cmake .. -DUSE_CUDA=ON
+  # For CUDA toolkit >= 11.4, `BUILD_WITH_CUDA_CUB` is required.
+  cmake .. -DUSE_CUDA=ON -DBUILD_WITH_CUDA_CUB=ON
  make -j4

 .. note:: Specifying compute capability