Missing some header files in Conda package

Ah, I solved this by setting some flags in various places. Now getting a different error here → PyBind11 doesn't recognize `torch._C.ScriptModule` as `torch::jit::Module`