[CI] optimze bazel compile guidance

.bazeliskrc

@@ -1 +1 @@
-USE_BAZEL_VERSION=5.3.1
+USE_BAZEL_VERSION=5.0.0

.bazelrc

@@ -15,9 +15,9 @@ build --host_cxxopt=-std=c++17
 build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
 
 # CUDA options
-build:cuda --action_env GCC_HOST_COMPILER_PATH="/opt/rh/devtoolset-9/root/usr/bin/gcc"
+build:cuda --action_env GCC_HOST_COMPILER_PATH="/usr/bin/gcc"
 build:cuda --action_env CUDA_TOOLKIT_PATH="/usr/local/cuda"
 build:cuda --action_env CUDA_VERSION="11"
 build:cuda --action_env CUDNN_VERSION="8"
-build:cuda --action_env CUDNN_INSTALL_PATH="/usr/local/cuda"
+build:cuda --action_env CUDNN_INSTALL_PATH="/usr/"
 build:cuda --action_env CUDA_COMPUTE_CAPABILITIES="7.5"

README.md

@@ -158,6 +158,13 @@ and also open for public contributions, bug fixes, and documentation. [[Contribu
 
 Basically, HierarchicalKV is a headers only library, the commands below only create binaries for benchmark and unit testing.
 
+Your environment must meet the following requirements:
+
+- CUDA version >= 11.2
+- NVIDIA GPU with compute capability 8.0, 8.6, 8.7 or 9.0
+- GCC supports `C++17' standard or later.
+- Bazel version >= 3.7.2 (Bazel compile only)
+
 ### with cmake
 ```shell
 git clone --recursive https://github.com/NVIDIA-Merlin/HierarchicalKV.git
@@ -176,20 +183,28 @@ For Unit Test:
 ```
 
 ### with bazel
+
+- DON'T use the option of `--recursive` for `git clone`.
+- Please modify the environment variables in the `.bazelrc` file in advance if using the customized docker images.
+- The docker images maintained on `nvcr.io/nvidia/tensorflow` are highly recommended.
+
+Pull the docker image:
 ```shell
-git clone --recursive https://github.com/NVIDIA-Merlin/HierarchicalKV.git
-cd HierarchicalKV && bazel build --config=cuda //...
+docker pull nvcr.io/nvidia/tensorflow:22.09-tf2-py3
+docker run --gpus all -it --rm nvcr.io/nvidia/tensorflow:22.09-tf2-py3
+```
+
+Compile in docker container:
+```shell
+git clone https://github.com/NVIDIA-Merlin/HierarchicalKV.git
+cd HierarchicalKV && bash bazel_build.sh
 ```
 
 For Benchmark:
 ```shell
 ./benchmark_util
 ```
 
-Your environment must meet the following requirements:
-
-- CUDA version >= 11.2
-- NVIDIA GPU with compute capability 8.0, 8.6, 8.7 or 9.0
 
 ## Benchmark & Performance(W.I.P)
 

build_deps/gpus/find_cuda_config.py

@@ -306,9 +306,11 @@ def _find_cublas_config(base_paths, required_version, cuda_version):
     if _at_least_version(cuda_version, "10.1"):
 
         def get_header_version(path):
-            version = (_get_header_version(path, name)
-                       for name in ("CUBLAS_VER_MAJOR", "CUBLAS_VER_MINOR",
-                                    "CUBLAS_VER_PATCH"))
+            version = (v for v in (_get_header_version(path, name)
+                                   for name in ("CUBLAS_VER_MAJOR",
+                                                "CUBLAS_VER_MINOR",
+                                                "CUBLAS_VER_PATCH",
+                                                "CUBLAS_VER_BUILD")) if v != "")
             return ".".join(version)
 
         header_path, header_version = _find_header(base_paths, "cublas_api.h",
@@ -338,9 +340,11 @@ def _find_cusolver_config(base_paths, required_version, cuda_version):
     if _at_least_version(cuda_version, "11.0"):
 
         def get_header_version(path):
-            version = (_get_header_version(path, name)
-                       for name in ("CUSOLVER_VER_MAJOR", "CUSOLVER_VER_MINOR",
-                                    "CUSOLVER_VER_PATCH"))
+            version = (v for v in (_get_header_version(path, name)
+                                   for name in ("CUSOLVER_VER_MAJOR",
+                                                "CUSOLVER_VER_MINOR",
+                                                "CUSOLVER_VER_PATCH",
+                                                "CUSOLVER_VER_BUILD")) if v != "")
             return ".".join(version)
 
         header_path, header_version = _find_header(base_paths,
@@ -369,9 +373,11 @@ def _find_curand_config(base_paths, required_version, cuda_version):
     if _at_least_version(cuda_version, "11.0"):
 
         def get_header_version(path):
-            version = (_get_header_version(path, name)
-                       for name in ("CURAND_VER_MAJOR", "CURAND_VER_MINOR",
-                                    "CURAND_VER_PATCH"))
+            version = (v for v in (_get_header_version(path, name)
+                                   for name in ("CURAND_VER_MAJOR",
+                                                "CURAND_VER_MINOR",
+                                                "CURAND_VER_PATCH",
+                                                "CURAND_VER_BUILD")) if v != "")
             return ".".join(version)
 
         header_path, header_version = _find_header(base_paths, "curand.h",
@@ -398,9 +404,11 @@ def _find_cufft_config(base_paths, required_version, cuda_version):
     if _at_least_version(cuda_version, "11.0"):
 
         def get_header_version(path):
-            version = (_get_header_version(path, name)
-                       for name in ("CUFFT_VER_MAJOR", "CUFFT_VER_MINOR",
-                                    "CUFFT_VER_PATCH"))
+            version = (v for v in (_get_header_version(path, name)
+                                   for name in ("CUFFT_VER_MAJOR",
+                                                "CUFFT_VER_MINOR",
+                                                "CUFFT_VER_PATCH",
+                                                "CUFFT_VER_BUILD")) if v != "")
             return ".".join(version)
 
         header_path, header_version = _find_header(base_paths, "cufft.h",
@@ -451,9 +459,11 @@ def _find_cusparse_config(base_paths, required_version, cuda_version):
     if _at_least_version(cuda_version, "11.0"):
 
         def get_header_version(path):
-            version = (_get_header_version(path, name)
-                       for name in ("CUSPARSE_VER_MAJOR", "CUSPARSE_VER_MINOR",
-                                    "CUSPARSE_VER_PATCH"))
+            version = (v for v in (_get_header_version(path, name)
+                                   for name in ("CUSPARSE_VER_MAJOR",
+                                                "CUSPARSE_VER_MINOR",
+                                                "CUSPARSE_VER_PATCH",
+                                                "CUSPARSE_VER_BUILD")) if v != "")
             return ".".join(version)
 
         header_path, header_version = _find_header(base_paths, "cusparse.h",