Havia um erro no README Estava escrito que não era necessário, mas o CUDA Toolkit é necessário.