Commit a772fd98 authored by viking1304's avatar viking1304

Update torch for ARM Macs to 2.3.1

parent a30b19dd
...@@ -16,7 +16,7 @@ export PYTORCH_ENABLE_MPS_FALLBACK=1 ...@@ -16,7 +16,7 @@ export PYTORCH_ENABLE_MPS_FALLBACK=1
if [[ "$(sysctl -n machdep.cpu.brand_string)" =~ ^.*"Intel".*$ ]]; then if [[ "$(sysctl -n machdep.cpu.brand_string)" =~ ^.*"Intel".*$ ]]; then
export TORCH_COMMAND="pip install torch==2.1.2 torchvision==0.16.2" export TORCH_COMMAND="pip install torch==2.1.2 torchvision==0.16.2"
else 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 fi
#################################################################### ####################################################################
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment