I’ll run it with 1.6.0 today. Data is unfortunately locked down but I’ll check on if the code is OK for me to share - I’m working on finding a minimal version that fails but haven’t had much luck yet. It’s a slow process since it takes so long to experience the error.
Also just FYI I also experience the error with a V100.
Thanks, I appreciate the help. I’ll also look into some of the new memory profiling tools in torch and see if they give any additional info.