RuntimeError: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error? Error 101: invalid device ordinal
The equipment I use is NVIDIA GeForce GTX TITAN X。
This error occurs when I run the function:
infer_helper.predict_dir function
RuntimeError: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error? Error 101: invalid device ordinal
The equipment I use is NVIDIA GeForce GTX TITAN X。
This error occurs when I run the function:
infer_helper.predict_dir function