当前位置:   article > 正文

PaddleOCR环境配置_paddleocr cuda

paddleocr cuda

PaddleOCR需要一些配置,这里主要记录cuDNN的配置。

亲测PaddleOCR需要cuDNNv7.x,CUDA11.0x。这里以cuDNN7.6为例

1.下载:cuDNNv7.6 cuDNN Archive | NVIDIA Developer  我的CUDA是11.0,但是使用cuDNNv7.6.0 for CUDA10.1也能用,很奇怪。

2.解压,打开所在文件夹,将里面的东西拷贝到CUDA根目录

  1. sudo cp cuda/include/cudnn.h /usr/local/cuda/include
  2. sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
  3. sudo chmod a+r /usr/local/cuda/include/cudnn.h
  4. sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

 3.接着再从终端打开

sudo gedit ~/.bashrc

4.将这句加到最后

export LD_LIBRARY_PATH = /usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH

5.最后

source ~/.bashrc

6.安装paddlepaddle-gpu

pip install paddlepaddle-gpu==2.0.1  -i https://mirrors.aliyun.com/pypi/simple/

7.验证paddlepaddle-gpu

  1. python
  2. import paddle
  3. paddle.utils.run_check()

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/喵喵爱编程/article/detail/946177
推荐阅读
相关标签
  

闽ICP备14008679号