Merge pull request #1024 from cmdr2/beta

Temp fix for macOS installations that installed torch 2.0 by mistake,…
This commit is contained in:
cmdr2 2023-03-17 13:28:15 +05:30 committed by GitHub
commit 38aafd3577
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -65,6 +65,11 @@ esac
# install torch and torchvision
if python ../scripts/check_modules.py torch torchvision; then
# temp fix for macOS installations that installed torch 2.0 by mistake
if [ "$OS_NAME" == "macos" ]; then
python -m pip install --upgrade torch==1.13.1 torchvision==0.14.1 -q
fi
echo "torch and torchvision have already been installed."
else
echo "Installing torch and torchvision.."