这个错误通常是因为安装时未指定使用的cpu版本。可以尝试在安装命令中指定使用GPU版本,例如:
pip install warp-rnnt[gpu]
如果您确实需要CPU版本,要使用最新版本的Warp-CTC和PyTorch,以免出现不兼容性。 如果您使用的是PyTorch 1.0或更高版本,您可以执行以下操作:
git clone https://github.com/HawkAaron/warp-rnnt.git
cd warp-rnnt
git submodule init
git submodule update
mkdir build
cd build/
cmake ..
make
pip install .
请注意,这也需要为CPU编译器安装OpenMP支持,例如,通过安装g ++,您可以在Ubuntu中安装OpenMP:
sudo apt-get install g++
然后,将Warp-CTC和PyTorch链接到Warp-RNNT的CMakeLists.txt中,如下所示:
add_subdirectory(pytorch_binding)
target_link_libraries(warp_rnnt PUBLIC warpctc torch)
然后重新编译Warp-RNNT:
cd warp-rnnt/
mkdir build_cpu
cd build_cpu/
cmake -Dpytorch_DIR=/share/cmake/PyTorch ../
make
将
替换为pytorch
的安装目录。 完成后,将Warp-RNNT PyTorch绑定安装到Python环境中:
cd ../pytorch_binding/
sudo python setup.py install
完成后,您应该可以在CPU上使用Warp-RNNT了。