From 95d5d7dbcc34672037b2da04aefa47c867e4352d Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Tue, 1 Jul 2025 00:27:34 -0700
Subject: [PATCH] Update setup.py

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

diff --git a/setup.py b/setup.py
index 59e9410674..b86de8c7e8 100644
--- a/setup.py
+++ b/setup.py
@@ -31,6 +31,8 @@ HAS_TORCH = False
 HAS_CUDA  = False
 HAS_HIP   = False
 torch     = None
+print(f"Unsloth: Imported torch with CUDA_HOME = {str(CUDA_HOME)}.")
+print(f"Unsloth: Imported torch with ROCM_HOME = {str(ROCM_HOME)}.")
 if importlib.util.find_spec("torch") is not None:
     try:
         import torch