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

Update cuda 11.7 nightly URL and add index.spec

Browse Source
This commit is contained in:
Anush Elangovan
2022-12-01 22:49:23 -08:00
parent 8856878cbd
commit 1106456651
5 changed files with 78 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
setup_venv.sh
View File

@@ -122,11 +122,11 @@ $PYTHON -m pip install --no-warn-conflicts -e . -f https://llvm.github.io/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
$PYTHON -m pip install --pre torch torchvision --extra-index-url https://download.pytorch.org/whl/nightly/cu117
if [ $? -eq 0 ];then
echo "Successfully Installed torch + cu116."
echo "Successfully Installed torch + cu117."
else
echo "Could not install torch + cu116." >&2
echo "Could not install torch + cu117." >&2
fi
fi