No valid algorithm found for EfficientNet and EfficientNetv2

[Deltadahl]

When I run the following code:

using Metalhead
using Flux

model = EfficientNet(:b0) |> gpu
data = rand(Float32, 224, 224, 3, 1) |> gpu
output = model(data)
println(output)

I get:

Warning: No valid algorithm found, probably bad params for convolution.
└ @ cuDNN C:\Users\ccsim\.julia\packages\cuDNN\3J08S\src\convolution.jl:280

The same thing happens even if I vary the input size (224, 224, 3, 1)-> (a, b, c, d), the model EfficientNet -> EfficientNetv2, the size of the model :b0 -> X. I have tried on Julia 1.9.1, 1.9.0 and 1.6.7 but I get the same Warning every time. I have also tried in new environments where I've only installed Metalhead and Flux.

Any help is much appreciated!