PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 pip3 install --no-cache-dir --verbose pycuda six

PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 echo "$PATH" &&
  echo "$LD_LIBRARY_PATH"

LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib64:

PATH=/usr/local/cuda/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 git clone --recursive -b ${TORCHAUDIO_VERSION} https://github.com/pytorch/audio torchaudio &&
  cd torchaudio &&
  sed -i 's#  URL https://zlib.net/zlib-1.2.11.tar.gz#  URL https://zlib.net/zlib-1.2.12.tar.gz#g' third_party/zlib/CMakeLists.txt ||
  echo "failed to patch torchaudio/third_party/zlib/CMakeLists.txt" &&
  sed -i 's#  URL_HASH SHA256=c3e5e9fdd5004dcb542feda5ee4f0ff0744628baf8ed2dd5d66f8ca1197cb1a1#  URL_HASH SHA256=91844808532e5ce316b3c010929493c0244f3d37593afd6de04f71821d5136d9#g' third_party/zlib/CMakeLists.txt ||
  echo "failed to patch torchaudio/third_party/zlib/CMakeLists.txt" &&
  BUILD_SOX=1 python3 setup.py install &&
  cd ../ &&
  rm -rf torchaudio

PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 pip3 install --no-cache-dir scikit-build &&
  pip3 install --no-cache-dir ninja &&
  pip3 install --no-cache-dir --verbose pysoundfile

PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 apt-get update &&
  apt-get install -y --no-install-recommends pkg-config libffi-dev libsndfile1 &&
  rm -rf /var/lib/apt/lists/* &&
  apt-get clean

TORCHAUDIO_VERSION

PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 PYTHON_ROOT=`pip3 show torch | grep Location: | cut -d' ' -f2` &&
  TORCH_CMAKE_CONFIG=$PYTHON_ROOT/torch/share/cmake/Torch/TorchConfig.cmake &&
  echo "patching _GLIBCXX_USE_CXX11_ABI in ${TORCH_CMAKE_CONFIG}" &&
  sed -i 's/  set(TORCH_CXX_FLAGS "-D_GLIBCXX_USE_CXX11_ABI=")/  set(TORCH_CXX_FLAGS "-D_GLIBCXX_USE_CXX11_ABI=0")/g' ${TORCH_CMAKE_CONFIG}

PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 cmake --version