Pytorch 0.3.0+ CUDA 8.0. When run the code after some epoches, it fails with the error information:
RuntimeError: cuda runtime error (4). I have tried some solutions including reboot the system, update pytorch, but it still has this error. Can someone help me solve this problem? Thanks a lot!