PyTorch bindings for Warp-ctc: c10/cuda/CUDAGuard.h: No such file or directory

I’m trying to install warp-CTC for deepSpeech2.

I’m using python 3.7.2, pytorch 1.1.0, cuda 9.0.

I get the following error:

src/binding.cpp:10:33: fatal error: c10/cuda/CUDAGuard.h: No such file or directory

I’d appreciate any direction for solving this, thanks.