RuntimeError: CUDA error: unspecified launch failure

pytorch训练时报错,是由于内存占用过多,占用率到99%了,把batch_size改小点就可以了,使得占用在90%左右就好

你可能感兴趣的:(RuntimeError: CUDA error: unspecified launch failure)