pytorch出现报错：
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)
将batch_size改小一点即可
我之前的
batch_size = 32,
改为
batch_size = 8
程序跑成功了。
出现上述问题的原因：
1.网络过于复杂（深）
2.电脑性能不行
版权声明：本文为weixin_45650259原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。