소스 검색

Use pip instead of conda for pytorch

oobabooga 2 년 전
부모
커밋
d2a7fac8ea
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      README.md

+ 2 - 2
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/.