v2

Author: allenwang28

.github/workflows/gpu_test.yaml

@@ -33,11 +33,27 @@ jobs:
         uses: actions/checkout@v4
       - name: Install enroot
         run: |
-          arch=$(dpkg --print-architecture)
-          curl -fSsL -O https://github.com/NVIDIA/enroot/releases/download/v3.5.0/enroot_3.5.0-1_${arch}.deb
-          curl -fSsL -O https://github.com/NVIDIA/enroot/releases/download/v3.5.0/enroot+caps_3.5.0-1_${arch}.deb
-          sudo apt-get install -y ./*.deb
-          rm -f enroot*.deb
+          arch=$(uname -m)
+          version="3.5.0"
+
+          # Install dependencies
+          sudo yum install -y jq squashfs-tools parallel || sudo apt-get install -y jq squashfs-tools parallel || true
+
+          # Download and install enroot
+          curl -fSsL -O https://github.com/NVIDIA/enroot/releases/download/v${version}/enroot-${version}-1.${arch}.rpm || \
+          curl -fSsL -O https://github.com/NVIDIA/enroot/releases/download/v${version}/enroot_${version}-1_amd64.deb
+
+          curl -fSsL -O https://github.com/NVIDIA/enroot/releases/download/v${version}/enroot+caps-${version}-1.${arch}.rpm || \
+          curl -fSsL -O https://github.com/NVIDIA/enroot/releases/download/v${version}/enroot+caps_${version}-1_amd64.deb
+
+          # Install based on available package manager
+          if command -v rpm &> /dev/null; then
+            sudo rpm -ivh enroot*.rpm
+          elif command -v dpkg &> /dev/null; then
+            sudo dpkg -i enroot*.deb
+          fi
+
+          rm -f enroot*
       - name: Setup conda env
         uses: conda-incubator/setup-miniconda@v2
         with: