upd

Author: yzh119

.github/workflows/nightly-release.yml

@@ -293,18 +293,12 @@ jobs:
           echo "RAM: $(free -h | awk '/^Mem:/ {print $7 " available out of " $2}')"
           echo "Disk: $(df -h / | awk 'NR==2 {print $4 " available out of " $2}')"
           echo "Architecture: $(uname -m)"
-          nvidia-smi
 
       - name: Checkout code
         uses: actions/checkout@v4
         with:
           submodules: true
 
-      - name: Set up Python
-        uses: actions/setup-python@v5
-        with:
-          python-version: '3.10'
-
       - name: Download flashinfer-python artifact
         uses: actions/download-artifact@v4
         with:

scripts/task_test_nightly_build.sh

@@ -12,6 +12,10 @@ set -x
 : ${DIST_JIT_CACHE_DIR:=dist-jit-cache}
 : ${DIST_PYTHON_DIR:=dist-python}
 
+# Display GPU information (running inside Docker container with GPU access)
+echo "=== GPU Information ==="
+nvidia-smi
+
 # Install flashinfer packages
 echo "Installing flashinfer-cubin from ${DIST_CUBIN_DIR}..."
 pip install ${DIST_CUBIN_DIR}/*.whl