训练网络的时候代码是可以跑通的,但是想要计算一下代码的大小和计算量就报了这个错误。解决的办法就是把显卡上的都放到CPU上面,就可以出结果了。至于为什么,我也不清楚,这么改能对。
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
最新推荐文章于 2024-05-23 10:38:16 发布
训练网络的时候代码是可以跑通的,但是想要计算一下代码的大小和计算量就报了这个错误。解决的办法就是把显卡上的都放到CPU上面,就可以出结果了。至于为什么,我也不清楚,这么改能对。