From 520fbcd163403db1a510f91cff3de4cb34c3f9be Mon Sep 17 00:00:00 2001 From: mrq Date: Thu, 16 Mar 2023 15:09:11 +0000 Subject: [PATCH] bumped torch up (CUDA: 11.8, ROCm, 5.4.2) --- setup-cuda.bat | 2 +- setup-cuda.sh | 2 +- setup-rocm.sh | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/setup-cuda.bat b/setup-cuda.bat index 405a9b6..76fd901 100755 --- a/setup-cuda.bat +++ b/setup-cuda.bat @@ -4,7 +4,7 @@ git submodule update --remote python -m venv venv call .\venv\Scripts\activate.bat python -m pip install --upgrade pip -python -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 +python -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118 python -m pip install -r .\modules\tortoise-tts\requirements.txt python -m pip install -e .\modules\tortoise-tts\ python -m pip install -r .\modules\dlas\requirements.txt diff --git a/setup-cuda.sh b/setup-cuda.sh index 81d6a9f..0e4c40e 100755 --- a/setup-cuda.sh +++ b/setup-cuda.sh @@ -7,7 +7,7 @@ python3 -m venv venv source ./venv/bin/activate python3 -m pip install --upgrade pip # just to be safe # CUDA -pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 +pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118 # install requirements python3 -m pip install -r ./modules/tortoise-tts/requirements.txt # install TorToiSe requirements python3 -m pip install -e ./modules/tortoise-tts/ # install TorToiSe diff --git a/setup-rocm.sh b/setup-rocm.sh index aa7dfa0..1e010e2 100755 --- a/setup-rocm.sh +++ b/setup-rocm.sh @@ -7,7 +7,7 @@ python3 -m venv venv source ./venv/bin/activate python3 -m pip install --upgrade pip # just to be safe # ROCM -pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2 +pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 # install requirements python3 -m pip install -r ./modules/tortoise-tts/requirements.txt # install TorToiSe requirements python3 -m pip install -e ./modules/tortoise-tts/ # install TorToiSe