minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Update Cutedsl version and pin cuda-python version (#16838)

Browse Source
This commit is contained in:
Baizhou Zhang
2026-01-10 17:08:43 +08:00
committed by GitHub
parent 3c358736d1
commit 9fd2358cc2
2 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+1 -2
View File
@@ -308,12 +308,11 @@ RUN --mount=type=cache,target=/root/.cache/pip \
RUN --mount=type=cache,target=/root/.cache/pip if [ "${CUDA_VERSION%%.*}" = "12" ]; then \ RUN --mount=type=cache,target=/root/.cache/pip if [ "${CUDA_VERSION%%.*}" = "12" ]; then \
python3 -m pip install nvidia-nccl-cu12==2.28.3 --force-reinstall --no-deps ; \ python3 -m pip install nvidia-nccl-cu12==2.28.3 --force-reinstall --no-deps ; \
python3 -m pip install nvidia-cudnn-cu12==9.16.0.29 --force-reinstall --no-deps ; \ python3 -m pip install nvidia-cudnn-cu12==9.16.0.29 --force-reinstall --no-deps ; \
python3 -m pip install nvidia-cutlass-dsl==4.3.0 --force-reinstall --no-deps ; \
elif [ "${CUDA_VERSION%%.*}" = "13" ]; then \ elif [ "${CUDA_VERSION%%.*}" = "13" ]; then \
python3 -m pip install nvidia-nccl-cu13==2.28.3 --force-reinstall --no-deps ; \ python3 -m pip install nvidia-nccl-cu13==2.28.3 --force-reinstall --no-deps ; \
python3 -m pip install nvidia-cublas==13.1.0.3 --force-reinstall --no-deps ; \ python3 -m pip install nvidia-cublas==13.1.0.3 --force-reinstall --no-deps ; \
python3 -m pip install nixl-cu13 ; \ python3 -m pip install nixl-cu13 ; \
python3 -m pip install nvidia-cutlass-dsl==4.3.1 --force-reinstall --no-deps ; \ python3 -m pip install cuda-python==13.1.1 ; \
fi fi
# Install development tools # Install development tools
python/pyproject.toml
+2 -2
View File
@@ -22,7 +22,7 @@ dependencies = [
"blobfile==3.0.0", "blobfile==3.0.0",
"build", "build",
"compressed-tensors", "compressed-tensors",
"cuda-python", "cuda-python==12.9",
"decord2", "decord2",
"datasets", "datasets",
"einops", "einops",
@@ -38,7 +38,7 @@ dependencies = [
"msgspec", "msgspec",
"ninja", "ninja",
"numpy", "numpy",
"nvidia-cutlass-dsl==4.2.1", "nvidia-cutlass-dsl==4.3.1",
"nvidia-ml-py", "nvidia-ml-py",
"openai-harmony==0.0.4", "openai-harmony==0.0.4",
"openai==2.6.1", "openai==2.6.1",