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