如何让EasyOCR调用GPU而非CPU?已装CUDA仍遇警告
让EasyOCR启用GPU的配置步骤
- 核对版本兼容性
先通过命令行输入nvcc --version查看你的CUDA版本,EasyOCR依赖PyTorch调用GPU,必须保证PyTorch版本与CUDA版本完全匹配,这是核心前提。 - 重装适配版PyTorch
先卸载现有PyTorch:
再根据你的CUDA版本安装对应PyTorch,比如CUDA 11.8就用这条命令:pip uninstall torch torchvision torchaudio
若CUDA是12.1版本,把命令里的pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118cu118替换成cu121即可。 - 验证PyTorch GPU可用性
运行以下代码:
输出import torch print(torch.cuda.is_available())True就说明PyTorch已成功识别GPU。 - 重新测试EasyOCR
再次运行你的EasyOCR初始化代码,CUDA not available的警告会消失,此时程序会自动使用GPU运行。 - 额外注意事项
确认你的GPU是NVIDIA型号(EasyOCR仅支持NVIDIA GPU),同时更新显卡驱动到最新版本,旧驱动可能导致CUDA无法被系统识别。
内容的提问来源于stack exchange,提问作者Julian Hessenberger
相关产品推荐
相关产品推荐

