Specify torchvision/torchaudio versions

TheTerrasque · Mar 18, 2023 · a0b1a30 · a0b1a30
1 parent c753261
commit a0b1a30
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -63,7 +63,7 @@ conda activate textgen
 
 | System | GPU | Command |
 |--------|---------|---------|
-| Linux/WSL | NVIDIA | `conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia` |
+| Linux/WSL | NVIDIA | `conda install torchvision=0.14.1 torchaudio=0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia` |
 | Linux | AMD | `pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2` |
 | MacOS + MPS (untested) | Any | `conda install pytorch torchvision torchaudio -c pytorch` |