ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

subprocess

Browse source
This commit is contained in:
Daniel Han 2025-07-01 00:51:04 -07:00
commit 2952513ba8
2 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

1
pyproject.toml
View file

@ -1,5 +1,4 @@
[build-system]
# Should be mirrored in requirements/build.txt
requires = [
"packaging>=24.2",
"setuptools>=77.0.3,<80.0.0",

5
setup.py
View file

@ -23,11 +23,10 @@ from setuptools import setup
from setuptools.command.install import install
ROOT_DIR = Path(__file__).parent
UNSLOTH_TARGET_DEVICE = os.environ.get("UNSLOTH_TARGET_DEVICE", "cuda")
IS_COLAB = "COLAB_" not in "".join(os.environ.keys())
import torch
from torch.utils.cpp_extension import CUDA_HOME, ROCM_HOME
output = subprocess.run(["python", "-c", "from torch.utils.cpp_extension import CUDA_HOME, ROCM_HOME; from torch.version import cuda, hip; print(CUDA_HOME); print(ROCM_HOME); print(cuda); print(hip);"], capture_output = True, text = True)
print(output)
# Try importing torch
HAS_TORCH = False