Hi,
whats your problem? have you searched the papers in this field?
there was this repo that I had used a year ago : https://github.com/kumar-shridhar/PyTorch-BayesianCNN 293
give that a try and it should get you going.
Update :
strongly recommend you to see : Implementing a Bayesian CNN in PyTorch 452