Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

(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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backends/trtllm/scripts/install_tensorrt.sh
View File

@ -34,6 +34,7 @@ install_ubuntu_requirements() {
if [ "$ARCH" = "aarch64" ];then ARCH="sbsa";fi 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 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 dpkg -i cuda-keyring_1.1-1_all.deb
rm /etc/apt/sources.list.d/cuda-ubuntu2404-x86_64.list
apt-get update apt-get update
if [[ $(apt list --installed | grep libcudnn9) ]]; then if [[ $(apt list --installed | grep libcudnn9) ]]; then