Skip to content

Commit

Permalink
Use pip instead of conda for pytorch
Browse files Browse the repository at this point in the history
  • Loading branch information
@oobabooga
oobabooga authored Mar 18, 2023
1 parent a0b1a30 commit d2a7fac
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -63,9 +63,9 @@ conda activate textgen

| System | GPU | Command |
|--------|---------|---------|
| Linux/WSL | NVIDIA | `conda install torchvision=0.14.1 torchaudio=0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia` |
| Linux/WSL | NVIDIA | `pip3 install torch torchvision torchaudio` |
| 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` |
| MacOS + MPS (untested) | Any | `pip3 install torch torchvision torchaudio` |

The up to date commands can be found here: https://pytorch.org/get-started/locally/.

Expand Down

0 comments on commit d2a7fac

Please sign in to comment.