fix(docs): add PyTorch to ROCm setup section
This commit is contained in:
parent
8834a6642f
commit
43787f085e
13
README.md
13
README.md
|
@ -49,7 +49,7 @@ Please [see the User Guide](https://github.com/ssube/onnx-web/blob/main/docs/use
|
|||
- [Note about setup paths](#note-about-setup-paths)
|
||||
- [Create a virtual environment](#create-a-virtual-environment)
|
||||
- [Install pip packages](#install-pip-packages)
|
||||
- [For AMD on Linux: Install ONNX ROCm](#for-amd-on-linux-install-onnx-rocm)
|
||||
- [For AMD on Linux: Install PyTorch and ONNX ROCm](#for-amd-on-linux-install-pytorch-and-onnx-rocm)
|
||||
- [For AMD on Windows: Install ONNX DirectML](#for-amd-on-windows-install-onnx-directml)
|
||||
- [For CPU on Linux: Install PyTorch CPU](#for-cpu-on-linux-install-pytorch-cpu)
|
||||
- [For CPU on Windows: Install PyTorch CPU](#for-cpu-on-windows-install-pytorch-cpu)
|
||||
|
@ -191,11 +191,14 @@ sure you are not using `numpy>=1.24`.
|
|||
[This SO question](https://stackoverflow.com/questions/74844262/how-to-solve-error-numpy-has-no-attribute-float-in-python)
|
||||
has more details.
|
||||
|
||||
#### For AMD on Linux: Install ONNX ROCm
|
||||
#### For AMD on Linux: Install PyTorch and ONNX ROCm
|
||||
|
||||
If you are running on Linux with an AMD GPU, download and install the ROCm version of `onnxruntime`:
|
||||
If you are running on Linux with an AMD GPU, download and install the ROCm version of `onnxruntime` and the ROCm
|
||||
version of PyTorch:
|
||||
|
||||
```shell
|
||||
> pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2
|
||||
|
||||
> wget https://download.onnxruntime.ai/onnxruntime_training-1.13.0.dev20221021001%2Brocm523-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
|
||||
|
||||
> pip install ./onnxruntime_training-1.13.0.dev20221021001%2Brocm523-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
|
||||
|
@ -251,9 +254,9 @@ If you are running with an Nvidia GPU on any operating system, install `onnxrunt
|
|||
PyTorch:
|
||||
|
||||
```shell
|
||||
> pip install onnxruntime-gpu
|
||||
> pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117
|
||||
|
||||
> pip install torch --extra-index-url https://download.pytorch.org/whl/cu117
|
||||
> pip install onnxruntime-gpu
|
||||
```
|
||||
|
||||
Make sure you have installed CUDA 11.x and that the version of PyTorch matches the version of CUDA
|
||||
|
|
Loading…
Reference in New Issue