diff --git a/docs/platforms/cpu_server.md b/docs/platforms/cpu_server.md
index 9e9e407ce..7e14b7a07 100644
--- a/docs/platforms/cpu_server.md
+++ b/docs/platforms/cpu_server.md
@@ -63,7 +63,7 @@ is required to enable SGLang service with CPU engine.
 conda create -n sgl-cpu python=3.12 -y
 conda activate sgl-cpu
 
-# Optional: Set PyTorch CPU as primary pip install channel to avoid installing CUDA version
+# Set PyTorch CPU as primary pip install channel to avoid installing the larger CUDA-enabled version and prevent potential runtime issues.
 pip config set global.index-url https://download.pytorch.org/whl/cpu
 pip config set global.extra-index-url https://pypi.org/simple
 
diff --git a/sgl-kernel/python/sgl_kernel/__init__.py b/sgl-kernel/python/sgl_kernel/__init__.py
index 81456dbbe..37ba4e3a9 100644
--- a/sgl-kernel/python/sgl_kernel/__init__.py
+++ b/sgl-kernel/python/sgl_kernel/__init__.py
@@ -23,7 +23,7 @@ def _find_cuda_home():
     return cuda_home
 
 
-if torch.version.hip is None:
+if torch.version.cuda is not None:
     cuda_home = Path(_find_cuda_home())
 
     if (cuda_home / "lib").is_dir():