apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(docs): add PyTorch to ROCm setup section

This commit is contained in:
Sean Sube 2023-01-24 22:30:46 -06:00
parent 8834a6642f
commit 43787f085e
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
README.md
View File

@ -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