当前位置: 首页 > 工具软件 > WARP-CTC > 使用案例 >

配置crnn的头文件warp-ctc的问题

杨飞飙
2023-12-01

记录以便大家解决。

warp-ctc安装

git clone https://github.com/SeanNaren/warp-ctc.git
cd warp-ctc
mkdir build; cd build
cmake ..
make

cd pytorch_binding
python setup.py install
1. 无法找到THC.h头文件-> pytorch版本过高

解决方法:不要使用现有的pytorch版本 都太新了,建议使用pytorch1.3.1及以下。
注:本人gcc version为7.2 前有的攻略都是5. 4.版本,但没有出对应的问题。

2. 安装yaml之后,重新进行warp_ctc的clone+编译即可

我没有出现需要在文件里加代码的情况,我在加入下面这个代码之后反而报错了。去掉这个代码按照原本的binging.cpp,install后成功。

下面是如果出现问题的可能解决方法:(修改pytorch_bingding/src/binding.cpp

1 at 92 line
int probs_size = THCudaTensor_size(state, probs, 2);
2 at l05 lines
void* gpu_workspace;
THCudaMalloc(state, &gpu_workspace, gpu_size_bytes);

3. 附注crnn.YAML文件

YAML文件配置方法(创建新环境)

文件名可命名为crnn.yaml 或其他。环境名对应name参数生成。

name: crnn
channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/   #清华源
  - https://mirrors.ustc.edu.cn/anaconda/pkgs/free/    #中科大源
  - rusty1s
  - pytorch
  - open3d-admin
  - nvidia
  - conda-forge
  - defaults
dependencies:
  - _libgcc_mutex=0.1=main
  - ca-certificates=2021.4.13=h06a4308_1
  - certifi=2020.12.5=py36h06a4308_0
  - ld_impl_linux-64=2.33.1=h53a641e_7
  - libffi=3.3=he6710b0_2
  - libgcc-ng=9.1.0=hdf63c60_0
  - libstdcxx-ng=9.1.0=hdf63c60_0
  - ncurses=6.2=he6710b0_1
  - openssl=1.1.1k=h27cfd23_0
  - pip=21.0.1=py36h06a4308_0
  - python=3.6.13=hdb3f193_0
  - readline=8.1=h27cfd23_0
  - setuptools=52.0.0=py36h06a4308_0
  - sqlite=3.35.4=hdfb4753_0
  - tk=8.6.10=hbc83047_0
  - wheel=0.36.2=pyhd3eb1b0_0
  - xz=5.2.5=h7b6447c_0
  - zlib=1.2.11=h7b6447c_3
  - pip:
    - absl-py==0.12.0
    - cachetools==4.2.1
    - chardet==4.0.0
    - colorama==0.4.4
    - cycler==0.10.0
    - google-auth==1.28.1
    - google-auth-oauthlib==0.4.4
    - grpcio==1.37.0
    - idna==2.10
    - importlib-metadata==3.10.1
    - kiwisolver==1.3.1
    - lmdb==0.98
    - markdown==3.3.4
    - matplotlib==3.3.4
    - nicelogger==2.0
    - nltk==3.3
    - numpy==1.18.2
    - oauthlib==3.1.0
    - opencv-python-headless==4.1.2.30
    - pillow==7.1.2
    - protobuf==3.15.8
    - pyasn1==0.4.8
    - pyasn1-modules==0.2.8
    - pyparsing==2.4.7
    - python-dateutil==2.8.1
    - requests==2.25.1
    - requests-oauthlib==1.3.0
    - rsa==4.7.2
    - six==1.14.0
    - tensorboard==2.4.1
    - tensorboard-plugin-wit==1.8.0
    - torch==1.3.1
    - torchvision==0.4.2
    - typing-extensions==3.7.4.3
    - urllib3==1.26.4
    - werkzeug==1.0.1
    - zipp==3.4.1
# prefix: / 可写地址可不写


 类似资料: