diff --git a/sgl-kernel/build.sh b/sgl-kernel/build.sh index 819283c51c4..65d4edc536a 100755 --- a/sgl-kernel/build.sh +++ b/sgl-kernel/build.sh @@ -13,7 +13,7 @@ if [ ${CUDA_VERSION} = "12.8" ]; then TORCH_INSTALL="pip install --no-cache-dir --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION//.}" else DOCKER_IMAGE="pytorch/manylinux-builder:cuda${CUDA_VERSION}" - TORCH_INSTALL="pip install --no-cache-dir torch==2.5.1 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}" + TORCH_INSTALL="pip install --no-cache-dir torch==2.6.0 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}" fi docker run --rm \ diff --git a/sgl-kernel/pyproject.toml b/sgl-kernel/pyproject.toml index cb5f388c2fb..916d4659038 100644 --- a/sgl-kernel/pyproject.toml +++ b/sgl-kernel/pyproject.toml @@ -1,7 +1,7 @@ [build-system] requires = [ "scikit-build-core>=0.10", - "torch>=2.5.1", + "torch>=2.6.0", "wheel", ] build-backend = "scikit_build_core.build" diff --git a/sgl-kernel/pyproject_rocm.toml b/sgl-kernel/pyproject_rocm.toml index dd122531b37..b72918b0eba 100644 --- a/sgl-kernel/pyproject_rocm.toml +++ b/sgl-kernel/pyproject_rocm.toml @@ -2,7 +2,7 @@ requires = [ "setuptools>=75.0", "scikit-build-core>=0.10", - "torch>=2.5.1", + "torch>=2.6.0", "wheel", ] build-backend = "setuptools.build_meta"