I am using a program that makes use of NudeNet to function. But I isolated this snippet of the code that is causing the issue. Every time when I start it, when it comes to calling the detector, it immediately just dumps that error message. I am not sure if it is an memory issue, cudnn just isn't being called by onnxruntime, or something else. Using a RTX 2060 (laptop).
**The code snippet which gave this error***
import os
import sys
import json
import subprocess
from nudenet import NudeDetector
detector = NudeDetector()
for filename in os.listdir(os.getcwd()):
if filename.endswith(".png"):
output = detector.detect(filename)
print(output)
Specify versions of the following libraries
nudenet == 2.0.9
tensorflow-gpu == 2.6.0
keras == 2.6.0
CUDA == 11.4
CUDNN == 8.2.4
Expected behavior
Output what class it believes the image is.
Describe the bug and error messages (if any) https://zerobin.net/?a226d1532fba7560#RiUihWfW4HG6t6D2OG6/MnUSd1wwhEO5ogRt7mKvIuw=
I am using a program that makes use of NudeNet to function. But I isolated this snippet of the code that is causing the issue. Every time when I start it, when it comes to calling the detector, it immediately just dumps that error message. I am not sure if it is an memory issue, cudnn just isn't being called by onnxruntime, or something else. Using a RTX 2060 (laptop).
**The code snippet which gave this error***
import os import sys import json import subprocess from nudenet import NudeDetector
detector = NudeDetector()
for filename in os.listdir(os.getcwd()): if filename.endswith(".png"): output = detector.detect(filename) print(output)
Specify versions of the following libraries
Expected behavior Output what class it believes the image is.