From 508be34294d296c07a07a0382a0b42dc7c9ca039 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 1 Jul 2025 00:32:23 -0700 Subject: [PATCH] Update setup.py --- setup.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) 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