Pytorch, INPUT (normal tensor) and WEIGHT (cuda tensor) mismatch
问题 DISCLAIMER I know, this question has already asked multiple times, but i tried their solutions, none of them worked for me, so after all those effort, i can't find anything else and eventually i have to ask again. I'm doing image classification with cnns (PYTORCH), i wan't to train it on GPU (nvidia gpu, compatible with cuda/cuda installed), i successfully managed to put net on it, but the problem is with data. if torch.cuda.is_available(): device = torch.device("cuda:0") print("Running on