Skip to content

Commit a0b1a30

Browse files
authored
Specify torchvision/torchaudio versions
1 parent c753261 commit a0b1a30

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ conda activate textgen
6363

6464
| System | GPU | Command |
6565
|--------|---------|---------|
66-
| Linux/WSL | NVIDIA | `conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia` |
66+
| Linux/WSL | NVIDIA | `conda install torchvision=0.14.1 torchaudio=0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia` |
6767
| Linux | AMD | `pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2` |
6868
| MacOS + MPS (untested) | Any | `conda install pytorch torchvision torchaudio -c pytorch` |
6969

0 commit comments

Comments
 (0)