Skip to content

Commit 2e26611

Browse files
README: add hint for installing PyTorch with ROCM on Linux
1 parent cdc8696 commit 2e26611

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

README.md

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,7 @@ sh ./run.sh
7373
```bash
7474
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
7575
```
76+
7677
3. Install the corresponding dependencies according to your own graphics card.
7778
- Nvidia GPU
7879
```bash
@@ -91,6 +92,11 @@ sh ./run.sh
9192
pip install -r requirements/ipex.txt
9293
```
9394

95+
4.If you are using an ROCM-capable AMD Radeon GPU, then you need to choose ROCM version of PyTorch.
96+
```bash
97+
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2
98+
```
99+
94100
## Preparation of Other Files
95101
### 1. Assets
96102
> RVC requires some models located in the `assets` folder for inference and training.
@@ -156,8 +162,8 @@ pacman -S rocm-hip-sdk rocm-opencl-sdk
156162
````
157163
For some models of graphics cards, you may need to configure the following environment variables (such as: RX6700XT).
158164
````
159-
export ROCM_PATH=/opt/rocm
160-
export HSA_OVERRIDE_GFX_VERSION=10.3.0
165+
export ROCM_PATH=/opt/rocm #Set ROCM Executables Path
166+
export HSA_OVERRIDE_GFX_VERSION=10.3.0 #Spoof GPU Model for ROCM
161167
````
162168
Also, make sure your current user is in the `render` and `video` user groups.
163169
````

0 commit comments

Comments
 (0)