Temp fix for macOS installations that installed torch 2.0 by mistake, when torch 2 released yesterday

This commit is contained in:
cmdr2 2023-03-17 13:27:44 +05:30
parent 0324deec60
commit a9da65c2cd

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.."