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。

Logo

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!

更多推荐