From 495e85bd58cc60cde763cff68830196e427ff1b1 Mon Sep 17 00:00:00 2001 From: Esteve Fernandez Date: Thu, 2 Nov 2023 15:29:19 +0100 Subject: [PATCH] build: cuDNN and TensorRT packages are installed in a separate Ansible script Signed-off-by: Esteve Fernandez --- ansible/roles/cuda/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ansible/roles/cuda/README.md b/ansible/roles/cuda/README.md index 1ab921b596d..59c4ac71410 100644 --- a/ansible/roles/cuda/README.md +++ b/ansible/roles/cuda/README.md @@ -26,7 +26,7 @@ wget https://developer.download.nvidia.com/compute/cuda/repos/$os/$(uname -m)/cu sudo dpkg -i cuda-keyring_1.1-1_all.deb sudo apt-get update cuda_version_dashed=$(eval sed -e "s/[.]/-/g" <<< "${cuda_version}") -sudo apt-get -y install cuda-${cuda_version_dashed} tensorrt-dev libcuddn8-dev +sudo apt-get -y install cuda-${cuda_version_dashed} ``` Perform the post installation actions: