diff --git a/atsetup.sh b/atsetup.sh index 61ff58a..6273911 100644 --- a/atsetup.sh +++ b/atsetup.sh @@ -291,7 +291,7 @@ install_custom_standalone() { echo " Downloading and installing PyTorch. This step can take a long time" echo " depending on your internet connection and hard drive speed. Please" echo " be patient." - pip install torch>=2.2.1+cu121 torchaudio>=2.2.1+cu121 --extra-index-url https://download.pytorch.org/whl/cu121 + pip install torch>=2.2.1+cu121 torchaudio>=2.2.1+cu121 --upgrade --force-reinstall --extra-index-url https://download.pytorch.org/whl/cu1 echo echo " Installing additional requirements." echo @@ -469,7 +469,7 @@ reapply_standalone() { echo " Downloading and installing PyTorch. This step can take a long time" echo " depending on your internet connection and hard drive speed. Please" echo " be patient." - pip install torch>=2.2.1+cu121 torchaudio>=2.2.1+cu121 --extra-index-url https://download.pytorch.org/whl/cu121 + pip install torch>=2.2.1+cu121 torchaudio>=2.2.1+cu121 --upgrade --force-reinstall --extra-index-url https://download.pytorch.org/whl/cu1 echo echo " Installing additional requirements." echo