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
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
####################################################################
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