forked from phoenix-oss/llama-stack-mirror
30 lines
637 B
Bash
30 lines
637 B
Bash
#!/bin/bash
|
|
|
|
if [[ $# -ne 1 ]]; then
|
|
echo "Error: Please provide the name of CONDA environment you wish to create"
|
|
exit 1
|
|
fi
|
|
|
|
ENV_NAME=$1
|
|
|
|
set -eu
|
|
eval "$(conda shell.bash hook)"
|
|
|
|
echo "Will build env (or overwrite) named '$ENV_NAME'"
|
|
|
|
set -x
|
|
|
|
run_build() {
|
|
# Set up the conda environment
|
|
yes | conda remove --name $ENV_NAME --all
|
|
yes | conda create -n $ENV_NAME python=3.10
|
|
conda activate $ENV_NAME
|
|
|
|
# PT nightly
|
|
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121
|
|
|
|
# install dependencies for `llama-agentic-system`
|
|
pip install -r fp8_requirements.txt
|
|
}
|
|
|
|
run_build
|