今天遇到了一个很无语的错,折腾了好久…用Bert提完特征网络用不了…检查了大小的问题,发现不是,后面发现是没有放在cuda上…
具体报错
RuntimeError: Caught RuntimeError in replica 0 on device 0.
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)
解决方法:
ext_hashCodes.unsqueeze(1).repeat(1, B, 1)
#改成
ext_hashCodes.unsqueeze(1).repeat(1, B, 1).cuda()
#就可以了