From c1bdba6bc99aed1071268caac1c394e81cf6775b Mon Sep 17 00:00:00 2001 From: weiqiang Date: Sat, 23 Sep 2023 22:23:19 +0000 Subject: [PATCH] Update setup-cuda.sh --- setup-cuda.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup-cuda.sh b/setup-cuda.sh index 72ffe14..c704b69 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/cu118 +pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu122 # 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