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)**
|
||||
|
||||
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
|
||||
pip uninstall onnxruntime onnxruntime-gpu
|
||||
|
|
|
@ -19,3 +19,4 @@ onnxruntime-gpu==1.17; sys_platform != 'darwin'
|
|||
tensorflow; sys_platform != 'darwin'
|
||||
opennsfw2==0.10.2
|
||||
protobuf==4.23.2
|
||||
pygrabber==0.2
|
||||
|
|
Loading…
Reference in New Issue