diff --git a/setup.py b/setup.py index cb1a79067..a66e3beb2 100644 --- a/setup.py +++ b/setup.py @@ -127,6 +127,8 @@ def check_cudnn_version_and_warn(global_option: str, required_cudnn_version: int version_ge_1_5 = ["-DVERSION_GE_1_5"] version_dependent_macros = version_ge_1_1 + version_ge_1_3 + version_ge_1_5 +raise_if_cuda_home_none("get_cuda_bare_metal_version") + _, bare_metal_version = get_cuda_bare_metal_version(CUDA_HOME) if "--distributed_adam" in sys.argv: