Hi @Ian
Did you find the cause of the issue? and how did you fix it? thanks
I am facing a similar issue 19. However, I get only the error:
RuntimeError: cuda runtime error (77) : an illegal memory access was encountered at /pytorch/aten/src/THC/generated/../THCTensorMathCompareT.cuh:69

when trying to access to a tensor. (such as printing it)
This issue seems to be related to the machine where 65 it is running.