Merge pull request #1025 from cmdr2/beta

temp fix for installations that installed torch 2.0 by mistake
This commit is contained in:
cmdr2 2023-03-17 13:33:41 +05:30 committed by GitHub
commit 700bfc16bd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -65,8 +65,10 @@ esac
# install torch and torchvision # install torch and torchvision
if python ../scripts/check_modules.py torch torchvision; then if python ../scripts/check_modules.py torch torchvision; then
# temp fix for macOS installations that installed torch 2.0 by mistake # temp fix for installations that installed torch 2.0 by mistake
if [ "$OS_NAME" == "macos" ]; then if [ "$OS_NAME" == "linux" ]; then
python -m pip install --upgrade torch==1.13.1+cu116 torchvision==0.14.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116 -q
elif [ "$OS_NAME" == "macos" ]; then
python -m pip install --upgrade torch==1.13.1 torchvision==0.14.1 -q python -m pip install --upgrade torch==1.13.1 torchvision==0.14.1 -q
fi fi