Got the solution? I have the same issue. A few days ago everything worked... Tried the same thing again and got cuda exception... Sorry, can't write a comment