记录以便大家解决。
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文件
文件名可命名为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: / 可写地址可不写