fix(reqs): bump to cuda 11.8

EveryVoiceTTS · Nov 11, 2023 · a035dbb · a035dbb
1 parent e828331
commit a035dbb
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
@@ -26,7 +26,7 @@ jobs:
           sudo apt-get install -y libsndfile1
       - name: Install dependencies and package
         run: |
-          CUDA_TAG=cu117 pip install -r requirements.torch.txt --find-links https://download.pytorch.org/whl/torch_stable.html
+          CUDA_TAG=cu118 pip install -r requirements.torch.txt --find-links https://download.pytorch.org/whl/torch_stable.html
           pip install cython
           pip install -e .
       - name: Install documentation dependencies

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -28,7 +28,7 @@ jobs:
           sudo apt-get install -y libsndfile1
       - name: Install dependencies and package
         run: |
-          CUDA_TAG=cu117 pip install -r requirements.torch.txt --find-links https://download.pytorch.org/whl/torch_stable.html
+          CUDA_TAG=cu118 pip install -r requirements.torch.txt --find-links https://download.pytorch.org/whl/torch_stable.html
           pip install -r requirements.test.txt
           pip install cython
           pip install -e .

diff --git a/readme.md b/readme.md
@@ -11,12 +11,12 @@ It is largely based off the FastSpeech2/Fastpitch models.
 
 ### Environment
 We recommend using Conda and Python 3.9. To do that, create a new environment,
-replacing cu117 (for CUDA 11.7) by your actual CUDA version's tag:
+replacing cu118 (for CUDA 11.8) by your actual CUDA version's tag (118 or higher):
 
 ```sh
 conda create --name EveryVoice python=3.9
 conda activate EveryVoice
-CUDA_TAG=cu117 pip install -r requirements.torch.txt --find-links https://download.pytorch.org/whl/torch_stable.html
+CUDA_TAG=cu118 pip install -r requirements.torch.txt --find-links https://download.pytorch.org/whl/torch_stable.html
 pip install cython
 ```
 

diff --git a/requirements.torch.txt b/requirements.torch.txt
@@ -1,4 +1,4 @@
 # these requirements have to be installed ahead of time in your environment and from a different URL:
-# CUDA_TAG=cu117 pip install -r requirements.torch.txt --find-links https://download.pytorch.org/whl/torch_stable.html
+# CUDA_TAG=cu118 pip install -r requirements.torch.txt --find-links https://download.pytorch.org/whl/torch_stable.html
 torch==2.1.0+${CUDA_TAG}
 torchaudio==2.1.0+${CUDA_TAG}