From d2a7fac8ea673ec29c82e41f68b50ac63a6c3d2e Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Sat, 18 Mar 2023 11:56:04 -0300 Subject: [PATCH] Use pip instead of conda for pytorch --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 407d8e4..078671a 100644 --- a/README.md +++ b/README.md @@ -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/.