作者在尝试使用自己的dataset进行十分类实验时遇到CUDA错误。首先，他们调整了可能过大的batch_size，然后确保模型被正确转移到GPU。最终，问题出在标签编码上，从1-10改为0-9后，错误得到解决。
一、实验设置
自己的dataset进行十分类实验，源代码的实验已成功跑通
二、bug
找到一个合适的代码之后，切换到自己的dataset，当图片格式完全匹配之后，报错：
RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling ‘cublasCreate(handle)’
三、解决过程
1.batch_size过大，memory占用太多–>调小batch_size到4，仍然行不通
2.在调用模型的时候没有model.cuda（）–>加上之后，行不通，仍然报错
3.stack overflow上相似错误显示是数据集标签的对应问题–>注意到自己的标签是从1-10，而应该是0-9，修改之后，成功！
扫一扫
评论 3
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
