github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-08 21:38:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

Install torch version required by torch-mlir when setting up importer venv. (#486)

Browse Source
This commit is contained in:
Ean Garvey
2022-11-14 02:31:01 -06:00
committed by GitHub
parent d9f4a9954a
commit 45677c1e23
10 changed files with 15 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
setup_venv.sh
View File

@@ -105,8 +105,6 @@ else
echo "Not installing a backend, please make sure to add your backend to PYTHONPATH"
fi
$PYTHON -m pip install -e . -f https://llvm.github.io/torch-mlir/package-index/ -f ${RUNTIME} -f https://download.pytorch.org/whl/nightly/torch/
if [[ ! -z "${IMPORTER}" ]]; then
echo "${Yellow}Installing importer tools.."
if [[ $(uname -s) = 'Linux' ]]; then
@@ -119,6 +117,9 @@ if [[ ! -z "${IMPORTER}" ]]; then
$PYTHON -m pip install --no-warn-conflicts --upgrade -r "$TD/requirements-importer-macos.txt" -f ${RUNTIME} --extra-index-url https://download.pytorch.org/whl/nightly/cpu
fi
fi
$PYTHON -m pip install --no-warn-conflicts -e . -f https://llvm.github.io/torch-mlir/package-index/ -f ${RUNTIME} -f https://download.pytorch.org/whl/nightly/torch/
if [[ $(uname -s) = 'Linux' && ! -z "${BENCHMARK}" ]]; then
$PYTHON -m pip uninstall -y torch torchvision
$PYTHON -m pip install --pre torch torchvision --extra-index-url https://download.pytorch.org/whl/nightly/cu116