在网上查了很多方案,在Torch社区也找了不少方案,但都对不上。
后来觉着这是CUDA的封装问题,将真正的错误信息掩盖了。
于是将程序改成在CPU上跑。
报错信息变成标签错误,进而发现是数据预处理打标签时错误。
修改相关代码后成功运行。
关于Python报错:RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCre`
最新推荐文章于 2024-08-01 21:22:04 发布
在网上查了很多方案,在Torch社区也找了不少方案,但都对不上。
后来觉着这是CUDA的封装问题,将真正的错误信息掩盖了。
于是将程序改成在CPU上跑。
报错信息变成标签错误,进而发现是数据预处理打标签时错误。
修改相关代码后成功运行。