almost working, now change to no cuda support