chengxy1998的博客

RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 23.70 GiB total capacity; 7.44 GiB already allocated; 87.88 MiB free; 7.71 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_.

** RuntimeError: Expected 4-dimensional input for 4-dimensional weight [10, 1, 5, 5], but got 3-dimensional input of size [1, 28, 28] instead ** 网络的第一层 此时我处理的是一张图片，而卷积神经网络模型是输入的是一个数据集的图片，而不是一张图片。 此时我的输入维度如下： 可通过如下方式来增加输入图片的维度： 增加维度过后： 至此，问题解决。 附注：最开始我