How to check if torch uses cuDNN - PyTorch Forums
discuss.pytorch.org › t › how-to-check-if-torch-usesJul 29, 2018 · So i just used packer to bake my own images for GCE and ran into the following situation. Installed CUDA 9.0 and everything worked fine, I could train my models on the GPU. Afte a while I noticed I forgot to install cuDNN, however it seems that pytorch does not complain about this. On an image with only CUDA installed, if I run torch.backends.cudnn.version() I get 7102 and torch.backends.cudnn ...