一、实验设置
自己的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,修改之后,成功!
bug记录:RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`
最新推荐文章于 2024-04-16 14:46:50 发布