一、背景
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博客