起初以为是torch版本的问题,后来发现是驱动没有编译。
具体参考了这两篇文章,博主写的非常好。
【一文解决】已安装CUDA与Pytorch但torch.cuda.is_available()为False
3分钟解决报错NVIDIA_SMI has failed because it couldn’t communicate with the NVIDIA driver.
起初以为是torch版本的问题,后来发现是驱动没有编译。
具体参考了这两篇文章,博主写的非常好。
【一文解决】已安装CUDA与Pytorch但torch.cuda.is_available()为False
3分钟解决报错NVIDIA_SMI has failed because it couldn’t communicate with the NVIDIA driver.