fix outdated cuda install

vespa-engine · Nov 24, 2023 · 5fd369c · 5fd369c
1 parent 4d6aad6
commit 5fd369c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/experimental/gpu-setup-rhel8-x86.sh b/experimental/gpu-setup-rhel8-x86.sh
@@ -35,8 +35,7 @@ FROM docker.io/vespaengine/vespa:latest
 USER root
 RUN dnf -y install 'dnf-command(config-manager)'
 RUN dnf -y config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo
-RUN dnf -y install libcudnn8-8.9.4.25-1.cuda11.8
-RUN dnf -y install vespa-onnxruntime-cuda
+RUN dnf -y install $(rpm -q --queryformat '%{NAME}-cuda-%{VERSION}' vespa-onnxruntime)
 USER vespa
 EOF
 

diff --git a/experimental/gpu-setup-rhel9-x86.sh b/experimental/gpu-setup-rhel9-x86.sh
@@ -35,8 +35,7 @@ FROM docker.io/vespaengine/vespa:latest
 USER root
 RUN dnf -y install 'dnf-command(config-manager)'
 RUN dnf -y config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo
-RUN dnf -y install libcudnn8-8.9.4.25-1.cuda11.8
-RUN dnf -y install vespa-onnxruntime-cuda
+RUN dnf -y install $(rpm -q --queryformat '%{NAME}-cuda-%{VERSION}' vespa-onnxruntime)
 USER vespa
 EOF