Skip to content

Commit 452442f

Browse files
authored
Update README.md
1 parent 9fb9c04 commit 452442f

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

README.md

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -80,14 +80,14 @@ python -m pip install --upgrade pip
8080
```
8181
* **Step 6** ➜ Install PyTorch with the appropriate "build:"
8282

83-
* **Windows/CUDA 11.8**```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118```
84-
* **Windows/CUDA 11.7**```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117```
85-
* **Windows/CPU-only**```pip install torch torchvision torchaudio```
86-
* **Linux/CUDA 11.8**```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118```
87-
* **Linux/CUDA 11.7**```pip install torch torchvision torchaudio```
88-
* **Linux/ROCm 5.4.2**```install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2```
89-
* **Linux/CPU-only**```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu```
90-
* **Apple/Metal/MPS**```pip pip install torch torchvision torchaudio```
83+
* **🪟Windows/CUDA 11.8**```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118```
84+
* **🪟Windows/CUDA 11.7**```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117```
85+
* **🪟Windows/CPU-only**```pip install torch torchvision torchaudio```
86+
* **🐧Linux/CUDA 11.8**```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118```
87+
* **🐧Linux/CUDA 11.7**```pip install torch torchvision torchaudio```
88+
* **🐧Linux/ROCm 5.4.2**```install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2```
89+
* **🐧Linux/CPU-only**```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu```
90+
* **🍎Apple/Metal/MPS**```pip pip install torch torchvision torchaudio```
9191
* **Metal/MPS** [speedup comparison](https://explosion.ai/blog/metal-performance-shaders) to a 5950x and RTX 3090.
9292

9393
* **Step 7** ➜ Doublecheck that you installed gpu-acceleration properly:

0 commit comments

Comments
 (0)