From 07e39696c56cd1f1010645c26601d8565c4d2cc3 Mon Sep 17 00:00:00 2001 From: Rothfive Date: Tue, 24 Oct 2023 02:01:49 +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 2c49c87..634b953 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 --index-url https://download.pytorch.org/whl/cu118 +pip3 install torch==2.1.0 torchvision torchaudio --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