1
0
Fork 0

fix(docs): note DirectML for Windows

This commit is contained in:
Sean Sube 2023-01-05 15:47:58 -06:00
parent 5a82f39cf8
commit 9a5ec9c4d6
1 changed files with 6 additions and 1 deletions

View File

@ -119,7 +119,12 @@ Install the following packages for AI:
> pip install numpy>=1.20,<1.24 # version is important, 1.24 removed the deprecated np.float symbol
> pip install accelerate diffusers transformers ftfy spacy scipy
> pip install onnx onnxruntime torch
> pip install onnxruntime-directml --force-reinstall # TODO: is this one necessary?
```
If you are running on Windows, install the DirectML ONNX runtime as well:
```shell
> pip install onnxruntime-directml --force-reinstall
```
Install the following packages for the web UI: