解决错误：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
·
在运行一个二分类的代码时候出现以下错误：
解决方法：
1.减小batch_size。检查GPU内存，这个错误很可能是由于GPU的运行内存不够引起的。
2.更新GPU驱动程序。
3.重启
4.检查硬件问题
5.上述方案都不能解决时候（我的问题就是这样...），尝试检查一下输出和标签的匹配问题。如标签的数量不等于网络输出通道的数量，即预测的输出类的数量。调整输出以匹配，它应该可以解决问题。
此外，可以使用如下命令格式来输出它的底层问题
CUDA_LAUNCH_BLOCKING=1 python [你的命令参数]
eg. CUDA_LAUNCH_BLOCKING=1 python train.py --task davis
新一代开源开发者平台 GitCode，通过集成代码托管服务、代码仓库以及可信赖的开源组件库，让开发者可以在云端进行代码托管和开发。旨在为数千万中国开发者提供一个无缝且高效的云端环境，以支持学习、使用和贡献开源项目。
更多推荐
2
0- 0
已为社区贡献2条内容
所有评论(0)