57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test script to check PaddleOCR GPU support and available parameters
|
|
"""
|
|
|
|
import paddleocr
|
|
import sys
|
|
|
|
def test_paddleocr_gpu():
|
|
print("Testing PaddleOCR GPU support...")
|
|
print(f"Python version: {sys.version}")
|
|
print(f"PaddleOCR version: {paddleocr.__version__}")
|
|
|
|
# Check available parameters by inspecting the PaddleOCR class
|
|
import inspect
|
|
sig = inspect.signature(paddleocr.PaddleOCR.__init__)
|
|
print(f"\nAvailable PaddleOCR parameters: {list(sig.parameters.keys())}")
|
|
|
|
# Try to initialize with different GPU parameters
|
|
print("\nTesting different GPU parameter combinations:")
|
|
|
|
# Test 1: Basic initialization (CPU)
|
|
try:
|
|
ocr1 = paddleocr.PaddleOCR(use_angle_cls=True, lang='en')
|
|
print("✓ Basic initialization (CPU) successful")
|
|
del ocr1
|
|
except Exception as e:
|
|
print(f"✗ Basic initialization failed: {e}")
|
|
|
|
# Test 2: Try use_gpu parameter
|
|
try:
|
|
ocr2 = paddleocr.PaddleOCR(use_angle_cls=True, lang='en', use_gpu=True)
|
|
print("✓ GPU initialization with use_gpu=True successful")
|
|
del ocr2
|
|
except Exception as e:
|
|
print(f"✗ GPU initialization with use_gpu=True failed: {e}")
|
|
|
|
# Test 3: Try gpu_id parameter
|
|
try:
|
|
ocr3 = paddleocr.PaddleOCR(use_angle_cls=True, lang='en', gpu_id=0)
|
|
print("✓ GPU initialization with gpu_id=0 successful")
|
|
del ocr3
|
|
except Exception as e:
|
|
print(f"✗ GPU initialization with gpu_id=0 failed: {e}")
|
|
|
|
# Test 4: Try enable_mkldnn parameter (Intel acceleration)
|
|
try:
|
|
ocr4 = paddleocr.PaddleOCR(use_angle_cls=True, lang='en', enable_mkldnn=True)
|
|
print("✓ CPU initialization with enable_mkldnn=True successful")
|
|
del ocr4
|
|
except Exception as e:
|
|
print(f"✗ CPU initialization with enable_mkldnn=True failed: {e}")
|
|
|
|
print("\nGPU configuration test completed.")
|
|
|
|
if __name__ == "__main__":
|
|
test_paddleocr_gpu() |