Hie, I built a simple CNN architecture when I try to run it, I get the error
RuntimeError: GET was unable to find an engine to execute this computation
Then I tried to find cudnn version using torch.backends.cudnn.version()
which throwed me another error
RuntimeError: cuDNN version incompatibility: PyTorch was compiled against (8, 5, 0) but found runtime version (8, 1, 0). PyTorch already comes bundled with cuDNN. One option to resolving this error is to ensure PyTorch can find the bundled cuDNN.Looks like your LD_LIBRARY_PATH contains incompatible version of cudnnPlease either remove it from the path or install cudnn (8, 5, 0)
.
But conda says my cudnn version is 8.5.0.96
A simple Linear Model works fine in GPU, but the CNN doesn’t. Below is the simple CNN I tried which throws me the error.
layer =nn.Conv2d(in_channels=1,
out_channels=10,
kernel_size=3,
stride=1,
padding=1).to(device)
rand_image_tensor = torch.randn(size=(1, 1, 28, 28)).to(device)
layer(rand_image_tensor).shape
PyTorch Version - 2.0.1+cu117
OS - WSL2
Ubuntu Version in WSL2: