remove cuda<0.6 constraint