Remove cuda version constraint.  Let it go back to 0.6