过于老旧的pytorch_ssim包 请从github下载源码

有些冷门算法真的不要随便pip,有可能下载到史前版本…最好还是找源代码 汗

今天要用到SSIM损失函数,从网上简单看了一下原理就想测试一下,偷了一下懒就直接在命令行输入pip install pytorch_ssim了,结果报了一堆错误(汗),正如这篇博文所说博文在此,传入Conv2D中的padding是浮点数,然而这样做完之后还会报错,差不多是说img和window不在一个运行设备上,我这边是window张量没有在gpu上,结果我用了.cuda()之后还是不管用,然后我就去github上看了一下,才发现这个pip包的版本太落后了。于是我重新下载覆盖之后,成功运行了!感谢!

点我github源码
过于落后的版本

后话:如果你想使用github上给出的案例,请你仿照知乎上这位给出的改进版本,源代码会报错。点这里

  • 10
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
根据引用中提供的内容,您可以使用以下命令来操作conda环境: - `conda env list`:查看环境列表 - `conda info --envs`:查看环境 - `conda remove --name env_name --all`:删除环境 - `conda remove --name env_name package_name`:删除环境中某个 - `conda deactivate`:关闭环境 如果您想更新pip,可以使用以下命令: - `python -m pip install --upgrade pip` 要升级numpy,可以运行以下命令: - `pip install -U numpy` 如果您想安装tensorflow GPU版本,可以使用以下命令: - `pip install tensorflow-gpu -i https://pypi.doubanio.com/simple` 要卸载tensorflow,可以运行以下命令: - `pip uninstall tensorflow tensorflow-gpu` 如果您想安装特定版本的tensorflow-gpu,可以使用以下命令: - `pip install tensorflow-gpu==2.4.0 -i https://pypi.doubanio.com/simple` 根据引用,当您遇到"tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape,要安装pytorch,您可以执行以下命令: - `conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge` 至于您遇到的"ModuleNotFoundError: No module named 'pytorch_ssim'"错误,这意味着您的环境中没有安装名为'pytorch_ssim'的模块。您可以尝试使用`pip install pytorch_ssim`命令来安装这个模块。如果仍然遇到问题,确保您已正确安装了所有必要的依赖项以及正确配置了您的环境。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* *3* [PyTorch深度学习(29)Pytorch code importance content](https://blog.csdn.net/jiangyangll/article/details/127601586)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 100%"] [ .reference_list ]
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值