Sorry for being late, @ptrblck but I have been extremely busy. The shapes of train_data and train_labels are (240000, 8), and (240000,), respectively. I installed PyTorch through pip, and my CUDA and cudnn versions are both 10.1. And, PyTorch version is 1.3.1. Feel free to ask any further information. Thanks.