Merge pull request #60 from vespa-engine/arnej/fix-old-cuda-install

fix outdated cuda install
vespa-engine · Nov 26, 2023 · c60d850 · c60d850
2 parents 4d6aad6 + 5fd369c
commit c60d850
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