RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
这个错误实际上可能是batch_size过大而导致的,因为有些人可能是用的A100-80G显存,而有些人可能只有24G-4090或者3090。此时只需要将batch_size改小或者换到大显存的卡上运行就可以了解决。
文章讨论了在运行时遇到CUDAerror:CUBLAS_STATUS_ALLOC_FAILED的问题,原因可能是batch_size过大导致显存不足,建议调整batch_size或升级显卡以解决问题。
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
这个错误实际上可能是batch_size过大而导致的,因为有些人可能是用的A100-80G显存,而有些人可能只有24G-4090或者3090。此时只需要将batch_size改小或者换到大显存的卡上运行就可以了解决。
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
2510
7392

被折叠的 条评论
为什么被折叠?