RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)
#8143
-
Hi, everyone. I have a repo whose link is https://github.com/zhoufengfan/pytorch-lightning-cifar10. When I run
But after I change the When I change the
The hardware parameters of my server are:
Can anyone help me to fix the bug? Thanks. |
Beta Was this translation helpful? Give feedback.
Replies: 3 comments 9 replies
-
This error can happen due to a number of reasons, could you try launching your script as: |
Beta Was this translation helpful? Give feedback.
-
However, after I change the
|
Beta Was this translation helpful? Give feedback.
-
OK, I know the reason. The reason is that I haven't installed NCCL on my machine. After I installed NCCL on my machine, the errors disappeared. Thanks to all the people who have replied to me. @awaelchli @SkafteNicki |
Beta Was this translation helpful? Give feedback.
OK, I know the reason. The reason is that I haven't installed NCCL on my machine.
After I installed NCCL on my machine, the errors disappeared.
Thanks to all the people who have replied to me. @awaelchli @SkafteNicki