解决错误：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
AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念，把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起，为开发者提供从开发、训练到部署的一站式体验。
更多推荐
2
0- 0
已为社区贡献2条内容
所有评论(0)