RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)
Exception raised from createCublasHandle at …\aten\src\ATen\cuda\CublasHandlePool.cpp:8 (most recent call first):

记录一下pytorch训练DCGAN的bug。。
找了半天才发现，原来是判别器最后一行没加Sigmoid，导致标签越界，超出范围[0,1]，所以报错了。