build: cuDNN and TensorRT packages are installed in a separate Ansibl…

…e script Signed-off-by: Esteve Fernandez <[email protected]>
autowarefoundation · Nov 2, 2023 · 495e85b · 495e85b
1 parent 8e932e1
commit 495e85b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: