chore: use torch 2.6 for sgl-kernel build (#5898)
This commit is contained in:
@@ -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 \
|
||||
|
||||
@@ -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"
|
||||
|
||||
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user