(misc): update dependency in trtllm dockerfile

This commit is contained in:
Morgan Funtowicz 2024-10-10 13:10:05 +00:00 committed by Morgan Funtowicz
parent cb69c9a967
commit 437c2aa142
1 changed files with 1 additions and 0 deletions

View File

@ -34,6 +34,7 @@ install_ubuntu_requirements() {
if [ "$ARCH" = "aarch64" ];then ARCH="sbsa";fi
curl -fsSLO https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/${ARCH}/cuda-keyring_1.1-1_all.deb
dpkg -i cuda-keyring_1.1-1_all.deb
rm /etc/apt/sources.list.d/cuda-ubuntu2404-x86_64.list
apt-get update
if [[ $(apt list --installed | grep libcudnn9) ]]; then