# Manually set cudnn convolution algorithm

**URL:** <https://discuss.pytorch.org/t/manually-set-cudnn-convolution-algorithm/101596>\
**Category:** vision\
**Created:** [November 4, 2020, 11:42am UTC](https://discuss.pytorch.org/t/manually-set-cudnn-convolution-algorithm/101596 "2020-11-04T11:42:37Z")\
**Posts on this page:** 1\
**Showing post:** 6

<div class="post-metadata">

**Author:** ![ptrblck](https://discuss.pytorch.org/user_avatar/discuss.pytorch.org/ptrblck/32/1823_2.png) [@ptrblck](https://discuss.pytorch.org/u/ptrblck)\
**Post date:** [September 24, 2021, 6:54pm UTC](https://discuss.pytorch.org/t/manually-set-cudnn-convolution-algorithm/101596/6 "2021-09-24T18:54:05Z")

</div>

A general cuDNN extension example can be found [here](https://github.com/pytorch/pytorch/blob/c371542efc31b1abfe6f388042aa3ab0cef935f2/test/cpp_extensions/cudnn_extension.cpp) and [this post](https://discuss.pytorch.org/t/implementing-a-custom-convolution-using-conv2d-input-and-conv2d-weight/18556/24) might also be useful.

---

_[View the full topic](https://discuss.pytorch.org/t/manually-set-cudnn-convolution-algorithm/101596)._