github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-09 13:57:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

Modify setup_venv to incorporate python version.

Browse Source
This commit is contained in:
Prashant Kumar
2022-04-18 06:11:21 +00:00
parent 8aed3ad5e2
commit d9f47b59f2
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
setup_venv.sh
View File

@@ -6,7 +6,7 @@
TD="$(cd $(dirname $0) && pwd)"
VENV_DIR="$TD/shark.venv"
if [ -z "$PYTHON" ]; then
PYTHON="$(which python)"
PYTHON="$(which python3)"
fi
echo "Setting up venv dir: $VENV_DIR"
@@ -25,11 +25,11 @@ source "$VENV_DIR/bin/activate" || die "Could not activate venv"
# reference to the python executable from the venv.
python -m pip install --upgrade pip || die "Could not upgrade pip"
python -m pip install --upgrade -r "$TD/requirements.txt"
pip install --find-links https://github.com/llvm/torch-mlir/releases torch-mlir
pip install --find-links https://github.com/NodLabs/SHARK/releases iree-compiler iree-runtime
pip install git+https://github.com/pytorch/functorch.git
pip install transformers
pip install .
python -m pip install --find-links https://github.com/llvm/torch-mlir/releases torch-mlir
python -m pip install --find-links https://github.com/NodLabs/SHARK/releases iree-compiler iree-runtime
python -m pip install git+https://github.com/pytorch/functorch.git
python -m pip install transformers
python -m pip install .
Red=`tput setaf 1`
Green=`tput setaf 2`