Try again

Author: NicolasHug

.github/scripts/setup-env.sh

@@ -100,7 +100,7 @@ echo '::endgroup::'
 
 echo '::group::Install torchvision-extra-decoders'
 # This can be done after torchvision was built
-if [[ "$(uname)" == "Linux" && "$ARCH" != "aarch64" ]]; then
+if [[ "$(uname)" == "Linux" && "$(uname -m)" != "aarch64" ]]; then
     extra_decoders_channel="--pre --index-url https://download.pytorch.org/whl/nightly/cpu"
 else
     extra_decoders_channel=""

packaging/post_build_script.sh

@@ -1,7 +1,7 @@
 #!/bin/bash
 LD_LIBRARY_PATH="/usr/local/lib:$CUDA_HOME/lib64:$LD_LIBRARY_PATH" python packaging/wheel/relocate.py
 
-if [[ "$(uname)" == "Linux" && "$ARCH" != "aarch64" ]]; then
+if [[ "$(uname)" == "Linux" && "$(uname -m)" != "aarch64" ]]; then
     extra_decoders_channel="--pre --index-url https://download.pytorch.org/whl/nightly/cpu"
 else
     extra_decoders_channel=""