有些冷门算法真的不要随便pip，有可能下载到史前版本…最好还是找源代码 汗
今天要用到SSIM损失函数，从网上简单看了一下原理就想测试一下，偷了一下懒就直接在命令行输入pip install pytorch_ssim了，结果报了一堆错误（汗），正如这篇博文所说博文在此，传入Conv2D中的padding是浮点数，然而这样做完之后还会报错，差不多是说img和window不在一个运行设备上，我这边是window张量没有在gpu上，结果我用了.cuda()之后还是不管用，然后我就去github上看了一下，才发现这个pip包的版本太落后了。于是我重新下载覆盖之后，成功运行了！感谢！
点我github源码
后话：如果你想使用github上给出的案例，请你仿照知乎上这位给出的改进版本，源代码会报错。点这里
