|
|
|
|
@@ -6,7 +6,7 @@ You can install SGLang using any of the methods below. For running DeepSeek V3/R
|
|
|
|
|
```
|
|
|
|
|
pip install --upgrade pip
|
|
|
|
|
pip install sgl-kernel --force-reinstall --no-deps
|
|
|
|
|
pip install "sglang[all]>=0.4.3.post1" --find-links https://flashinfer.ai/whl/cu124/torch2.5/flashinfer-python
|
|
|
|
|
pip install "sglang[all]>=0.4.3.post2" --find-links https://flashinfer.ai/whl/cu124/torch2.5/flashinfer-python
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
Note: SGLang currently uses torch 2.5, so you need to install the flashinfer version for torch 2.5. If you want to install flashinfer separately, please refer to [FlashInfer installation doc](https://docs.flashinfer.ai/installation.html). Please note that the package currently used by FlashInfer is named `flashinfer-python`, not `flashinfer`.
|
|
|
|
|
@@ -19,7 +19,7 @@ If you experience an error like `OSError: CUDA_HOME environment variable is not
|
|
|
|
|
## Method 2: From source
|
|
|
|
|
```
|
|
|
|
|
# Use the last release branch
|
|
|
|
|
git clone -b v0.4.3.post1 https://github.com/sgl-project/sglang.git
|
|
|
|
|
git clone -b v0.4.3.post2 https://github.com/sgl-project/sglang.git
|
|
|
|
|
cd sglang
|
|
|
|
|
|
|
|
|
|
pip install --upgrade pip
|
|
|
|
|
@@ -35,7 +35,7 @@ Note: To AMD ROCm system with Instinct/MI GPUs, do following instead:
|
|
|
|
|
|
|
|
|
|
```
|
|
|
|
|
# Use the last release branch
|
|
|
|
|
git clone -b v0.4.3.post1 https://github.com/sgl-project/sglang.git
|
|
|
|
|
git clone -b v0.4.3.post2 https://github.com/sgl-project/sglang.git
|
|
|
|
|
cd sglang
|
|
|
|
|
|
|
|
|
|
pip install --upgrade pip
|
|
|
|
|
@@ -63,7 +63,7 @@ docker run --gpus all \
|
|
|
|
|
Note: To AMD ROCm system with Instinct/MI GPUs, it is recommended to use `docker/Dockerfile.rocm` to build images, example and usage as below:
|
|
|
|
|
|
|
|
|
|
```bash
|
|
|
|
|
docker build --build-arg SGL_BRANCH=v0.4.3.post1 -t v0.4.3.post1-rocm630 -f Dockerfile.rocm .
|
|
|
|
|
docker build --build-arg SGL_BRANCH=v0.4.3.post2 -t v0.4.3.post2-rocm630 -f Dockerfile.rocm .
|
|
|
|
|
|
|
|
|
|
alias drun='docker run -it --rm --network=host --device=/dev/kfd --device=/dev/dri --ipc=host \
|
|
|
|
|
--shm-size 16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined \
|
|
|
|
|
@@ -72,11 +72,11 @@ alias drun='docker run -it --rm --network=host --device=/dev/kfd --device=/dev/d
|
|
|
|
|
drun -p 30000:30000 \
|
|
|
|
|
-v ~/.cache/huggingface:/root/.cache/huggingface \
|
|
|
|
|
--env "HF_TOKEN=<secret>" \
|
|
|
|
|
v0.4.3.post1-rocm630 \
|
|
|
|
|
v0.4.3.post2-rocm630 \
|
|
|
|
|
python3 -m sglang.launch_server --model-path meta-llama/Llama-3.1-8B-Instruct --host 0.0.0.0 --port 30000
|
|
|
|
|
|
|
|
|
|
# Till flashinfer backend available, --attention-backend triton --sampling-backend pytorch are set by default
|
|
|
|
|
drun v0.4.3.post1-rocm630 python3 -m sglang.bench_one_batch --batch-size 32 --input 1024 --output 128 --model amd/Meta-Llama-3.1-8B-Instruct-FP8-KV --tp 8 --quantization fp8
|
|
|
|
|
drun v0.4.3.post2-rocm630 python3 -m sglang.bench_one_batch --batch-size 32 --input 1024 --output 128 --model amd/Meta-Llama-3.1-8B-Instruct-FP8-KV --tp 8 --quantization fp8
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
## Method 4: Using docker compose
|
|
|
|
|
|