其他分享
首页 > 其他分享> > PyTorch踩坑记录:torch.ger / torch.outer算子与torch库版本的关系

PyTorch踩坑记录:torch.ger / torch.outer算子与torch库版本的关系

作者:互联网

前情提要:在npu服务器上运行程序,ger算子报错,查看pytorch官方文档后将torch.ger换成了torch.outer

在这里插入图片描述

程序运行报错:显示torch没有outer算子

在这里插入图片描述
查看torch版本发现我的torch版本是1.5.0

在这里插入图片描述
关键点来了:

在pytorch1.7.0之后,torch.ger算子被删掉了,用torch.outer代替。

也就是说,只有pytorch1.7.0的torch.outer与torch.ger功能等价,因此torch1.5.0没有outer算子。

于是我安装了最新的torch1.8.1,当然原来的torch1.5.0也自动被uninstalled了

在这里插入图片描述
安装好之后,再次运行程序,没有报错,问题解决。

标签:ger,outer,torch,pytorch1.7,报错,算子
来源: https://blog.csdn.net/qq_43799400/article/details/118995634