我的模型无法在GPU上运行,我发现这是安装过程中的错误。 该文件说:
否则,将WARP_CTC_PATH设置为安装libwarpctc.so的任何位置。如果您使用的是GPU,则还应确保将CUDA_HOME设置为home cuda目录(即include / cuda.h和lib / libcudart.so所在的目录)。例如:
导出CUDA_HOME =“ / usr / local / cuda”
但是我应该在哪里设置CUDA_HOME ??? 它在.bashrc中吗?但是我已经设定好了!该模型仍未在GPU上运行。