From 032e979bdc52acf81a10145452a74186a7583f6f Mon Sep 17 00:00:00 2001
From: x0152 <45021974+x0152@users.noreply.github.com>
Date: Wed, 5 Apr 2023 21:05:25 +0300
Subject: [PATCH] Update setup.py

---
 setup.py | 2 ++
 1 file changed, 2 insertions(+)

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: