[solved] Is CUDNN7 group convolution supported


I am creating resnext and xception models on pytorch. Both models use grouped convolution which is supported in CUDNN 7. Does current pytorch support this? Thanks.

1 Like

If you build PyTorch from source and have cuDNN 7, you’ll get fast grouped convolution support in cuDNN. PyTorch v0.2.0 uses a slower for-loop over each group.

Thanks for the reply. The latest source support cudnn7 for group convolution. I will build it.