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