diff --git a/atsetup.sh b/atsetup.sh index 070ef83..0b458da 100644 --- a/atsetup.sh +++ b/atsetup.sh @@ -325,7 +325,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 --upgrade --force-reinstall --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/cu121 echo echo " Installing additional requirements." echo