install: support 24.04 containers. remove cudnn8 and cuda11.

This commit is contained in:
Koushik Dutta
2024-10-09 11:11:51 -07:00
parent f7ccdf0795
commit 196316ad97

View File

@@ -23,7 +23,6 @@ then
&& wget -qO /cuda-keyring.deb https://developer.download.nvidia.com/compute/cuda/repos/$distro/$(uname -m)/cuda-keyring_1.1-1_all.deb \
&& dpkg -i /cuda-keyring.deb \
&& apt update -q \
&& apt install -y cuda-nvcc-11-8 libcublas-11-8 libcudnn8 cuda-libraries-11-8 \
&& apt install -y cuda-nvcc-12-6 libcublas-12-6 libcudnn9-cuda-12 cuda-libraries-12-6;
if [ "$?" != "0" ]