|
|
@ -93,8 +93,8 @@ AMD users can install rocm and pytorch with pip if you don't have it already ins |
|
|
|
|
|
|
|
|
|
|
|
```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.4.2``` |
|
|
|
```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.4.2``` |
|
|
|
|
|
|
|
|
|
|
|
This is the command to install the nightly with ROCm 5.5 that supports the 7000 series and might have some performance improvements: |
|
|
|
This is the command to install the nightly with ROCm 5.6 that supports the 7000 series and might have some performance improvements: |
|
|
|
```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.5 -r requirements.txt``` |
|
|
|
```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.6 -r requirements.txt``` |
|
|
|
|
|
|
|
|
|
|
|
### NVIDIA |
|
|
|
### NVIDIA |
|
|
|
|
|
|
|
|
|
|
|