UserWarning: CUDA initialization: CUDA driver initialization failed, you might not have a CUDA gpu.
I created a pod with A40 GPU but I am getting the following error when running torch.cuda.is_available()
2 Replies
Unknown User•8mo ago
Message Not Public
Sign In & Join Server To View
I think it was a bug. I recreated the pod after some hours and it got solved.
I was not using any temperature. I was just creating the pod and then running
torch.cuda.is_available() to see if the driver is working or not.