Skip to content

Commit

Permalink
Wildcard does not work for debianversion.
Browse files Browse the repository at this point in the history
Query repo instead.
  • Loading branch information
xkszltl committed Jun 19, 2024
1 parent 392e958 commit 2de56bc
Showing 1 changed file with 15 additions and 2 deletions.
17 changes: 15 additions & 2 deletions pkgs/cuda.sh
Original file line number Diff line number Diff line change
Expand Up @@ -127,8 +127,21 @@
'debian' | 'linuxmint' | 'ubuntu')
sudo DEBIAN_FRONTEND=noninteractive apt-get -o 'DPkg::Lock::Timeout=3600' install --allow-downgrades -y \
{"cudnn9-cuda-$CUDA_VER_MAJOR",libcudnn9-samples} \
libnccl{2,-dev}"=*-*+cuda$CUDA_VER_MAJOR.*" \
"tensorrt=10.*-*+cuda$CUDA_VER_MAJOR.*"
$(
set -e +x >/dev/null
for i in libnccl{2,-dev} tensorrt; do
apt-cache -o 'DPkg::Lock::Timeout=3600' show "$i" \
| grep -e'^'{Package,Version}':' \
| sed -n 's/^[^:]*:[[:space:]]*//p' \
| sed 's/[[:space:]]*$//' \
| paste -d= - - \
| grep -v '[[:space:]]' \
| grep '^[^=][^=]*=.*\-.*\+cuda'"$CUDA_VER_MAJOR"'\.[0-9][0-9]*$' \
| sort -V \
| tail -n1 \
| grep .
done
)
;;
esac
ldconfig -p | grep libcudnn
Expand Down

0 comments on commit 2de56bc

Please sign in to comment.