chore: bump sgl-kernel 0.3.0 with torch 2.8.0 (#8718)
This commit is contained in:
@@ -23,10 +23,10 @@ fi
|
||||
|
||||
if [ ${CUDA_VERSION} = "12.8" ]; then
|
||||
DOCKER_IMAGE="${BUILDER_NAME}:cuda${CUDA_VERSION}"
|
||||
TORCH_INSTALL="pip install --no-cache-dir torch==2.7.1 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}"
|
||||
TORCH_INSTALL="pip install --no-cache-dir torch==2.8.0 --index-url https://download.pytorch.org/whl/test/cu128"
|
||||
else
|
||||
DOCKER_IMAGE="${BUILDER_NAME}:cuda${CUDA_VERSION}"
|
||||
TORCH_INSTALL="pip install --no-cache-dir torch==2.7.1"
|
||||
TORCH_INSTALL="pip install --no-cache-dir torch==2.8.0 --index-url https://download.pytorch.org/whl/test/cu126"
|
||||
fi
|
||||
|
||||
docker run --rm \
|
||||
|
||||
Reference in New Issue
Block a user