Solved by ptrblck in post #2 
              
              
                Depending where the bottleneck in your current system is, ~5 minutes might be expected. I just reran the script on a V100 server (which should be of course a bit more powerful) and it finished in 34s. 


This post gives you a good overview about potential bottlenecks and their workarounds. 


This…