pytorch出现报错:
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)
将batch_size改小一点即可
我之前的
batch_size = 32,
改为
batch_size = 8
程序跑成功了。
出现上述问题的原因:
1.网络过于复杂(深)
2.电脑性能不行
pytorch出现报错:
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)
将batch_size改小一点即可
我之前的
batch_size = 32,
改为
batch_size = 8
程序跑成功了。
出现上述问题的原因:
1.网络过于复杂(深)
2.电脑性能不行