Skip to content

Commit

Permalink
Add PyTorch 2.4 support (#339)
Browse files Browse the repository at this point in the history
  • Loading branch information
rusty1s authored Aug 15, 2024
1 parent e5cae67 commit 601c8bf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .github/workflows/cuda/Linux.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 601c8bf

Please sign in to comment.