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

RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling cublasCreate(handle)``出现这个问题可能有很多原因,如果版本检查pytorch和CUDA匹配的话,一定检查一下你的网络结构设置的正不正确,有没有初始化参数写错、tensor大小不相容的错误,亲测有效,我的网络初始化需要一个参数改变最后隐藏层的数目,因为忘记设置,数据大小和网络不匹配,就会报这个错!

你可能感兴趣的:(pytorch,深度学习,神经网络)