Skip to content

Commit cd161c6

Browse files
authored
Update README.md
1 parent 76cb126 commit cd161c6

File tree

1 file changed

+8
-19
lines changed

1 file changed

+8
-19
lines changed

README.md

Lines changed: 8 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
</tr>
2323
<tr>
2424
<td>AMD</td>
25-
<td>⚠️ (see below)</td>
25+
<td></td>
2626
<td>✅</td>
2727
<td>ROCm 5.6</td>
2828
</tr>
@@ -49,7 +49,7 @@ You must install the these before following the installation instructions below:
4949
<summary>🪟WINDOWS INSTRUCTIONS🪟</summary>
5050

5151
### Step 1
52-
🟢 Nvidia GPU ➜ Install [CUDA 11.8](https://developer.nvidia.com/cuda-11-8-0-download-archive)
52+
🟢 Nvidia GPU ➜ Install [CUDA 11.8](https://developer.nvidia.com/cuda-11-8-0-download-archive)<br>
5353
🔴 AMD GPU - Unfortunately, PyTorch does not currently support AMD GPUs on Windows. It's only supported on Linux. There are several ways to possibly get around this limitation, but I'm unable to verify since I don't have an AMD GPU. See [HERE](https://ubuntu.com/tutorials/install-ubuntu-on-wsl2-on-windows-11-with-gui-support#1-overview), [HERE](https://ubuntu.com/tutorials/enabling-gpu-acceleration-on-ubuntu-on-wsl2-with-the-nvidia-cuda-platform#1-overview), and possibly [HERE](https://user-images.githubusercontent.com/108230321/275660295-e2d6e097-38c5-4e38-9a1f-f28441ba8812.png).
5454
### Step 2
5555
Download the ZIP file from the latest "release," unzip anywhere on your computer, and go into the ```src``` folder.
@@ -65,23 +65,11 @@ Activate the virtual environment:
6565
```
6666
### Step 5
6767
```
68-
python -m pip install --upgrade pip
69-
```
70-
### Step 6
71-
🟢 Nvidia GPUs:
72-
```
73-
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
74-
```
75-
🔴 AMD GPUs - To reiterate, PyTorch does not supprot AMD GPUs Windows, only Linux.<br><br>
76-
🔵 CPU only:
77-
```
78-
pip install torch torchvision torchaudio
79-
```
80-
### Step 7
81-
```
82-
pip install -r requirements.txt
68+
python setup.py
8369
```
84-
### Optional Step 8 - Double check GPU-Acceleration
70+
> And just follow the instructions.
71+
72+
### Optional Step 6 - Double check GPU-Acceleration
8573
Run this script if you want to doublecheck that you installed the Pytorch and gpu-acceleration software correctly:
8674
```
8775
python check_gpu.py
@@ -92,9 +80,10 @@ python check_gpu.py
9280
<summary>🐧LINUX INSTRUCTIONS🐧</summary>
9381

9482
### Step 1
95-
🟢 Nvidia GPUs ➜ Install [CUDA 11.8](https://developer.nvidia.com/cuda-11-8-0-download-archive)
83+
🟢 Nvidia GPUs ➜ Install [CUDA 11.8](https://developer.nvidia.com/cuda-11-8-0-download-archive)<br>
9684
🔴 AMD GPUs ➜ Install [ROCm version 5.6](https://docs.amd.com/en/docs-5.6.0/deploy/windows/gui/index.html).
9785
> [THIS REPO](https://github.com/nktice/AMD-AI) might also help if AMD's instructions aren't clear.
86+
9887
### Step 2
9988
Download the ZIP file from the latest "release," unzip anywhere on your computer, and go into the ```src``` folder.
10089
### Step 3

0 commit comments

Comments
 (0)