Update torch for ARM Macs to 2.3.1

This commit is contained in:
viking1304 2024-06-20 23:57:59 +02:00
parent a30b19dd55
commit a772fd9804

View File

@ -16,7 +16,7 @@ export PYTORCH_ENABLE_MPS_FALLBACK=1
if [[ "$(sysctl -n machdep.cpu.brand_string)" =~ ^.*"Intel".*$ ]]; then
export TORCH_COMMAND="pip install torch==2.1.2 torchvision==0.16.2"
else
export TORCH_COMMAND="pip install torch==2.3.0 torchvision==0.18.0"
export TORCH_COMMAND="pip install torch==2.3.1 torchvision==0.18.1"
fi
####################################################################