[backport] Support building with CTK11.5. (#7379) (#7411)

* Support building with CTK11.5. * Require system cub installation for CTK11.4+. * Check thrust version for segmented sort.
2021-11-10 19:23:29 +08:00
parent e7ac2486eb
commit 11f8b5cfcd
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