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