shiyi099

Pytorch <2.1.0 can't load the checkpoints correctly

1

When I try to use lower version of Pytorch such as 2.0.0 ,1.13.0 and so on，the checkpoint "bootstapir_checkpoint.pt" doesn't work correctly. The error shows below： --------------------------------------------------------------------------- RuntimeError Traceback (most recent...

None of the algorithms provided by cuDNN heuristics worked; trying fallback algorithms

2

My configure of Python package is jax == 0.4.13 jaxlib == 0.4.13+cu12+cudnn89 My Hardware is Nvidia-H800 On Ubuntu20.04 LTS x86_64 When I try to run inference of TAPNET or TAPIR...