From 843734d2968be3547d4717d8018290410dbbeb4e Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Tue, 1 Jul 2025 00:19:25 -0700
Subject: [PATCH] Update setup.py

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

diff --git a/setup.py b/setup.py
index b411d4d18c..192aea7b7f 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,8 @@ if importlib.util.find_spec("torch") is not None:
         HAS_TORCH = True
         HAS_CUDA  = (torch.version.cuda is not None)
         HAS_HIP   = (torch.version. hip is not None)
-
+        print(f"Unsloth: Imported torch with CUDA_HOME = {str(CUDA_HOME)}.")
+        print(f"Unsloth: Imported torch with ROCM_HOME = {str(ROCM_HOME)}.")
     except Exception as e:
         print(f"Unsloth: Importing torch failed with error = {str(e)}.")
         HAS_TORCH = False