zxybazh commented on issue #13639: URL: https://github.com/apache/tvm/issues/13639#issuecomment-1374403257
Yes I can see the same warning `CUDA driver initialization failed, you might not have a CUDA gpu.` I can also get `torch.cuda.is_available()` returning `True` for me. When using your script, I don't see the same Cuda free error. My pytorch with cuda support is installed via pip so that could probably be the difference here. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
