diff --git a/setup.py b/setup.py
index b86de8c7e8..192aea7b7f 100644
--- a/setup.py
+++ b/setup.py
@@ -28,11 +28,9 @@ IS_COLAB = "COLAB_" not in "".join(os.environ.keys())
 
 # Try importing torch
 HAS_TORCH = False
-HAS_CUDA  = False
+HAS_CUDA  = True
 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