diff --git a/slurm/README.md b/slurm/README.md index 101064594..662adc14a 100644 --- a/slurm/README.md +++ b/slurm/README.md @@ -5,7 +5,7 @@ conda create -n sglang124 python=3.11 conda activate sglang124 -pip install torch=2.5.1 --index-url https://download.pytorch.org/whl/cu124 +pip install torch==2.5.1 --index-url https://download.pytorch.org/whl/cu124 pip install sgl-kernel --force-reinstall --no-deps pip install "sglang[all]>=0.4.2.post4" --find-links https://flashinfer.ai/whl/cu124/torch2.5/flashinfer/ @@ -27,4 +27,4 @@ python scripts/generate_reasoning.py \ --num-generations 2 \ --max-tokens 16384 \ --max-concurrent 200 -``` \ No newline at end of file +```