为了用EAST大模型测试图片中字体方向,需要使用torch和opencv,在使用中发现速度太慢,研究了一下如何使用GPU加速(N卡):
- 下载安装 NVIDIA CUDA下载页面
- 卸载原有的opencv:pip uninstall opencv-python opencv-contrib-python -y
- 从github上下载支持CUDA的 opencv-python: https://github.com/cudawarped/opencv-python-cuda-wheels/releases,并安装
- 从NVIDIA官网下载cuDNN
- 测试