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