python运行报错***
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)
***解决办法:
1、将所设的batch_size改小一点。因为是从github上copy的代码,所以跑不通其实挺正常,毕竟电脑配置啥的,都与作者的不一样。因此我们改成和自己电脑相匹配的就行。
将batch_size改成32,就成功运行。
大家遇到这种情况根据自身情况,改小一点即可。
2、根据自己的数据集改成相应的num_classes.
eg.你的数据集为CIFAR100,有100类,则num_classes=100。
更多推荐