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

错误解决方法

排查一:

把网络的各个shape打印出来,发现其实是数据shape不匹配。

有问题先去看数据、embedding、网络维度!!!

排查二:

如果shape都是对应的,但是还是报相同的错误,应该是GPU内存不够,可以换一个空闲的GPU。或者调小Batch都可以。

你可能感兴趣的:(深度学习,人工智能,python)