@ptrblck
thank you for your reply.
I only run my program on two selected gpus.
I’ve tested your suggestion, but nothing changes.
Honestly, I didn’t explicitly check whether the data is feed the model, BUT, the for-loop used for the training process just doesn’t iterate, so I guess the data is not fed into the model.
The for-loop is not iterating. After consuming hundreds of MB of GPU memory, the program seems to be freezed.
I found another piece of information a few minutes ago, which might be helpful.
I know, it’s 2018, but the lab server is still using CUDA 7.5, but I’m using PyTorch 0.4.0 compiled with CUDA 8.0.
Please don’t ask me to switch back to PyTorch 0.3.0 with CUDA 7.5. I’ve tried, and there are a lot of horrible compatibility issues.