Compare commits
5 Commits
50de6481a0
...
8cfafa31fc
Author | SHA1 | Date |
---|---|---|
|
8cfafa31fc | |
|
ec6d7d2995 | |
|
e791f2f18a | |
|
2663c7eae7 | |
|
aafeb8cf89 |
|
@ -188,7 +188,10 @@ pip install -r requirements.txt
|
||||||
**CUDA Execution Provider (Nvidia)**
|
**CUDA Execution Provider (Nvidia)**
|
||||||
|
|
||||||
1. Install [CUDA Toolkit 11.8.0](https://developer.nvidia.com/cuda-11-8-0-download-archive)
|
1. Install [CUDA Toolkit 11.8.0](https://developer.nvidia.com/cuda-11-8-0-download-archive)
|
||||||
2. Install dependencies:
|
2. Install [cuDNN v8.9.7 for CUDA 11.x](https://developer.nvidia.com/rdp/cudnn-archive) (required for onnxruntime-gpu):
|
||||||
|
- Download cuDNN v8.9.7 for CUDA 11.x
|
||||||
|
- Make sure the cuDNN bin directory is in your system PATH
|
||||||
|
3. Install dependencies:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
pip uninstall onnxruntime onnxruntime-gpu
|
pip uninstall onnxruntime onnxruntime-gpu
|
||||||
|
|
|
@ -19,3 +19,4 @@ onnxruntime-gpu==1.17; sys_platform != 'darwin'
|
||||||
tensorflow; sys_platform != 'darwin'
|
tensorflow; sys_platform != 'darwin'
|
||||||
opennsfw2==0.10.2
|
opennsfw2==0.10.2
|
||||||
protobuf==4.23.2
|
protobuf==4.23.2
|
||||||
|
pygrabber==0.2
|
||||||
|
|
Loading…
Reference in New Issue