warp_ctc安装期间的GPU支持

时间:2018-08-04 08:40:29

标签: pytorch

我的模型无法在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上运行。

0 个答案:

没有答案