【新方案】RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
问题使用GPU训练时报错,报错内容如下/pytorch/aten/src/ATen/native/cuda/Loss.cu:247:nll_loss_forward_reduce_cuda_kernel_2d:block:[0,0,0],thread:[15,0,0]Assertion`t>=0&&t=0&&t=0&&t=0&&t=0&&t=0&&t=0&&t=0&&t=0&&t=0&&t=0&&