easyocr 警告 CUDA 不可用 - 默认为 CPU。注意：使用 GPU 时该模块速度更快

Question

我参考这篇文章来检查我的cuda驱动程序。就满足要求了。但torch仍然无法使用GPU。

操作系统信息： Windows 10

包和环境信息：

(.env) PS E:\TMP> nvidia-smi
Tue Dec 20 14:25:12 2022       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 512.77       Driver Version: 512.77       CUDA Version: 11.6     |

(.env) PS E:\TMP> python
Python 3.10.4 (tags/v3.10.4:9d38120, Mar 23 2022, 23:13:41) [MSC v.1929 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
False

(.env) PS E:\TMP> pip freeze
...
easyocr==1.6.2
opencv-python==4.6.0.66
opencv-python-headless==4.5.4.60
torch==1.13.1
torchaudio==0.13.1+cu116
torchvision==0.14.1

任何帮助将不胜感激。

Answer 1

目前 Windows 上的 PyTorch 仅支持 Python 3.7-3.9；不支持 Python 2.x。

参考

将python版本更改为3.8即可工作。

Answer 2

我也有同样的问题。搜索了一段时间后，我在JaidedAI/EasyOCR github存储库上发现了以下注释：

对于 Windows，请先按照此处的官方说明安装 torch 和 torchvision https://pytorch.org。在 pytorch 网站上，请务必选择您拥有的正确的 CUDA 版本。如果您打算仅在 CPU 模式下运行，请选择

CUDA = None

。

卸载torch和easyOCR后，我先重新安装torch，然后重新安装easyOCR。看来问题已经解决了。

希望这有帮助。