install in check

RachelZheng · Mar 15, 2023 · eb0042c · eb0042c
1 parent 7d9b53d
commit eb0042c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/release_build.yml b/.github/workflows/release_build.yml
@@ -165,7 +165,7 @@ jobs:
         conda run -n build_binary python --version
     - name: Install C/C++ compilers
       run: |
-        sudo yum install -y gcc gcc-c++
+        sudo yum install -y gcc gcc-c++ libnvidia-ml-dev
     - name: Install PyTorch and CUDA
       shell: bash
       run: |