From 601c8bf16703e601c90e393687bb6c0939747f09 Mon Sep 17 00:00:00 2001 From: Matthias Fey Date: Thu, 15 Aug 2024 22:43:37 +0200 Subject: [PATCH] Add PyTorch 2.4 support (#339) --- .github/workflows/cuda/Linux.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/cuda/Linux.sh b/.github/workflows/cuda/Linux.sh index 42a14f18..a035e940 100755 --- a/.github/workflows/cuda/Linux.sh +++ b/.github/workflows/cuda/Linux.sh @@ -62,7 +62,7 @@ sudo mv cuda-${OS}.pin /etc/apt/preferences.d/cuda-repository-pin-600 wget -nv ${URL}/${FILENAME} sudo dpkg -i ${FILENAME} -if [ "${1}" = "cu117" ] || [ "${1}" = "cu118" ] || [ "${1}" = "cu121" || [ "${1}" = "cu124" ]; then +if [ "${1}" = "cu117" ] || [ "${1}" = "cu118" ] || [ "${1}" = "cu121" ] || [ "${1}" = "cu124" ]; then sudo cp /var/cuda-repo-${APT_KEY}/cuda-*-keyring.gpg /usr/share/keyrings/ else sudo apt-key add /var/cuda-repo-${APT_KEY}/7fa2af80.pub