README: add instructions for installing ROCM version of PyTorch

This commit is contained in:
Jax Leo 2025-01-28 06:27:52 +00:00
parent 865e4eacaf
commit ab620b6023
5 changed files with 31 additions and 6 deletions

View File

@ -170,9 +170,15 @@ pacman -S rocm-hip-sdk rocm-opencl-sdk
You might also need to set these environment variables (e.g. on a RX6700XT): You might also need to set these environment variables (e.g. on a RX6700XT):
```` ````
export ROCM_PATH=/opt/rocm export ROCM_PATH=/opt/rocm #Set ROCM Executables Path
export HSA_OVERRIDE_GFX_VERSION=10.3.0 export HSA_OVERRIDE_GFX_VERSION=10.3.0 #Spoof GPU Model for ROCM
```` ````
And overwrite PyTorch with its ROCM version after installing dependencies.
````
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2
````
Make sure your user is part of the `render` and `video` group: Make sure your user is part of the `render` and `video` group:
```` ````
sudo usermod -aG render $USERNAME sudo usermod -aG render $USERNAME

View File

@ -80,7 +80,7 @@ pip install -r requirements-dml.txt
pip install -r requirements-ipex.txt pip install -r requirements-ipex.txt
# Cartes AMD sur Linux (ROCm) # Cartes AMD sur Linux (ROCm)
pip install -r requirements-amd.txt pip install -r requirements.txt
``` ```
------ ------
@ -151,6 +151,10 @@ Vous devrez peut-être créer ces variables d'environnement (par exemple avec RX
export ROCM_PATH=/opt/rocm export ROCM_PATH=/opt/rocm
export HSA_OVERRIDE_GFX_VERSION=10.3.0 export HSA_OVERRIDE_GFX_VERSION=10.3.0
```` ````
Et remplacer PyTorch par sa version ROCM après l'installation des dépendances.
````
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2
````
Assurez-vous que votre utilisateur est dans les groupes `render` et `video`: Assurez-vous que votre utilisateur est dans les groupes `render` et `video`:
```` ````
sudo usermod -aG render $USERNAME sudo usermod -aG render $USERNAME

View File

@ -100,7 +100,7 @@ pip install -r requirements-dml.txt
- A カード ROCM(Linux) - A カード ROCM(Linux)
```bash ```bash
pip install -r requirements-amd.txt pip install -r requirements.txt
``` ```
- I カード IPEX(Linux) - I カード IPEX(Linux)
@ -202,6 +202,11 @@ export ROCM_PATH=/opt/rocm
export HSA_OVERRIDE_GFX_VERSION=10.3.0 export HSA_OVERRIDE_GFX_VERSION=10.3.0
``` ```
そして、依存関係をインストールした後に、PyTorchをROCMバージョンで上書きする。
````
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2
````
同時に、現在のユーザーが`render`および`video`ユーザーグループに属していることを確認してください。 同時に、現在のユーザーが`render`および`video`ユーザーグループに属していることを確認してください。
``` ```

View File

@ -100,7 +100,7 @@ pip install -r requirements-dml.txt
- A카드ROCM(Linux) - A카드ROCM(Linux)
```bash ```bash
pip install -r requirements-amd.txt pip install -r requirements.txt
``` ```
- I카드IPEX(Linux) - I카드IPEX(Linux)
@ -202,6 +202,12 @@ export ROCM_PATH=/opt/rocm
export HSA_OVERRIDE_GFX_VERSION=10.3.0 export HSA_OVERRIDE_GFX_VERSION=10.3.0
``` ```
그리고 종속 요소를 설치한 후 PyTorch를 ROCM 버전으로 덮어씁니다.
````
pip 설치 토치 토치비전 토치오디오 --index-url https://download.pytorch.org/whl/rocm6.2
````
동시에 현재 사용자가 `render``video` 사용자 그룹에 속해 있는지 확인하세요. 동시에 현재 사용자가 `render``video` 사용자 그룹에 속해 있는지 확인하세요.
``` ```

View File

@ -89,7 +89,7 @@ for Intel ARC graphics cards on Linux / WSL using Python 3.10:
pip install -r requirements-ipex.txt pip install -r requirements-ipex.txt
for AMD graphics cards on Linux (ROCm): for AMD graphics cards on Linux (ROCm):
pip install -r requirements-amd.txt pip install -r requirements.txt
``` ```
------ ------
@ -165,6 +165,10 @@ Talvez você também precise definir estas variáveis de ambiente (por exemplo,
export ROCM_PATH=/opt/rocm export ROCM_PATH=/opt/rocm
export HSA_OVERRIDE_GFX_VERSION=10.3.0 export HSA_OVERRIDE_GFX_VERSION=10.3.0
```` ````
E substitui o PyTorch pela sua versão ROCM depois de instalar as dependências.a
````
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2
````
Verifique também se seu usuário faz parte do grupo `render` e `video`: Verifique também se seu usuário faz parte do grupo `render` e `video`:
```` ````
sudo usermod -aG render $USERNAME sudo usermod -aG render $USERNAME