diff --git a/.github/workflows/torchao.yml b/.github/workflows/torchao.yml
index c2bf850ea5..d5d6016aa1 100644
--- a/.github/workflows/torchao.yml
+++ b/.github/workflows/torchao.yml
@@ -29,6 +29,7 @@ jobs:
           sudo nvidia-smi -pm 1
           sudo nvidia-smi -ac 1215,1410
           nvidia-smi
+          sudo ldconfig
       - name: Clone and setup conda env
         run: |
           CONDA_ENV=${BASE_CONDA_ENV} . "${SETUP_SCRIPT}"
@@ -43,7 +44,7 @@ jobs:
           if [ -d .userbenchmark ]; then rm -Rf .userbenchmark; fi
           # Install torchao
           echo "Installing torchao"
-          python install.py torchao
+          python install.py --userbenchmark torchao
           echo "Running the torchao userbenchmark"
           python run_benchmark.py torchao --ci
           cp -r ./.userbenchmark/torchao ../benchmark-output