theseus库中functorch包和torch版本兼容问题解决

一、背景

theseus库的导入中要用到functorch,这个包在torch1.13后不再单独作为一个包,而是集成在torch内部,通过torch.func使用。

plantf代码要求torch版本1.12

functorch适配的torch1.12,官方说明版本为0.2.0,但是实际上并不兼容。

二、解决办法

functorch指定0.2.0办法失败;

1.直接升级torch到2.0,然后找到对应的torchvision和chaudio版本:

(这样就不用再下载functorch拉)

pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1

 2.还会出现natten的问题,重新下载一个。参考如下。

参考:

PyTorch中torch、torchvision、torchaudio、torchtext版本对应关系_torch与torchvision版本_Liekkas Kono的博客-CSDN博客

NATTEN

https://pytorch.org/functorch/versions.html

  • 7
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值