Solved by eqy in post #7 
              
              
                You could consider opening an issue on the github for this. I am curious if there is a workaround that works like computing a cruder hash function using functions that are native on GPU so that only a very small amount of data has to be copied to CPU for hashing: 
import time
import torch

class Has…