RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
CUBLAS 是 CUDA 专门用来解决线性代数运算的库。而上述报错则是说明读取的数据类别数与网络模型中设置的类别数不一致造成。比如,你的数据是5各类别,而你的网络中的分类数(num_classes)设置成了6。首先,你要检查读取数据的路径看是否是正确读取想要的数据;对比是否labels的最大值与读取数据的类别数一致。然后,打印网络中的标签(即labels);
原创
2023-03-20 13:08:04 ·
2013 阅读 ·
0 评论