CUDA Support for Bazel¶
Introduction¶
Since there is no native CUDA support in Bazel (Ref: Bazel Issue #6578: Native CUDA Support), the Authors of Apollo borrowed CUDA support for Bazel from TensorFlow project.
Status¶
Support upto CUDA Toolkit 11.1 and cuDNN8
How to generate find_cuda_config.py.gz.base64
¶
cd ${APOLLO_ROOT_DIR}/third_party/gpus/
python3 compress_find_cuda_config.py
Notes¶
Please note that CUDA support for Bazel in Apollo was tailored to run on Linux ONLY. Windows and MacOS support were stripped off which are available in the original TensorFlow project.
As for CPU architecture, only x86_64
and aarch64(arm64)
support was
reserved.