Commit 733117ea authored by Woosuk Kwon's avatar Woosuk Kwon
Browse files

Add build script

parent cb02853e
#!/bin/bash
eval "$(conda shell.bash hook)"
PYTORCH_VERSION="2.2.1"
for PYTHON_VERSION in 38 39 310 311; do
source ~/.bashrc;
conda activate vllm-flash-py${PYTHON_VERSION};
conda env list;
pip install packaging ninja;
pip install torch==${PYTORCH_VERSION};
time python setup.py bdist_wheel --dist-dir=dist;
done
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment