install libcudnn8 for cuda 11.8

NVIDIA-Merlin · Jun 5, 2023 · 5d4c6e7 · 5d4c6e7
1 parent e2fbc8e
commit 5d4c6e7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/gpu-tests.yml b/.github/workflows/gpu-tests.yml
@@ -27,7 +27,7 @@ jobs:
       - name: Install Ubuntu packages
         run: |
           apt-get update -y
-          apt-get install -y build-essential protobuf-compiler git lsb-release libcudnn8
+          apt-get install -y build-essential protobuf-compiler git lsb-release 'libcudnn8=*cuda11.8'
       - name: Set up Python 3.8
         uses: actions/setup-python@v4
         with: