Update torch for macos to have arm64

OpenGATE · Jul 23, 2024 · 3c5ca6c · 3c5ca6c
1 parent 052dbf6
commit 3c5ca6c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -129,8 +129,8 @@ jobs:
               wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-1.10.1%2Bcpu.zip
               unzip libtorch-cxx11-abi-shared-with-deps-1.10.1+cpu.zip
             elif [ "${{ matrix.os }}" == 'macos-latest' ]; then
-              wget https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.10.1.zip
-              unzip libtorch-macos-1.10.1.zip
+              wget https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.2.0.zip
+              unzip libtorch-macos-arm64-2.2.0.zip
             fi
           fi
           if [ "${{ matrix.options }}" == 'rtk' ]; then