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

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

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

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

PyTorch踩坑记录:torch.ger / torch.outer算子与torch库版本的关系
查看torch版本发现我的torch版本是1.5.0

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

在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了

PyTorch踩坑记录:torch.ger / torch.outer算子与torch库版本的关系
安装好之后,再次运行程序,没有报错,问题解决。

上一篇:2021-05-10


下一篇:Java编程思想学习笔记(七)