Hey, sorry because you use the in-function version, try that:
import torch.nn as nn
drop      = nn.Dropout2d()
then, you use drop as a function like: drop(…)
By the way which CuDNN version did you use?