feat: add blackwell Dockerfile (#5302)

2025-04-11 13:08:53 -07:00
parent 60bcbf2a35
commit f774a0d275
2 changed files with 30 additions and 0 deletions
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -57,6 +57,17 @@ srt = [
    "einops",
 ]

+blackwell = [
+    "sglang[runtime_common]",
+    "sgl-kernel",
+    "torch",
+    "torchvision",
+    "cuda-python",
+    "outlines>=0.0.44,<=0.1.11",
+    "partial_json_parser",
+    "einops",
+]
+
 # HIP (Heterogeneous-computing Interface for Portability) for AMD
 # => base docker rocm/vllm-dev:20250114, not from public vllm whl
 srt_hip = [