关于权重加载运行

[421212-huv]

torch和 torchvision的版本号方便说一下吗，我加载模型的时候总出现不匹配问题

[shenyunhang]

以下的版本试过是没问题的：

torch==2.6.0
torchvision==0.20.0
transformers==4.49.0

也可以看下是不是模型加载错了。

[421212-huv]

现在遇见了问题，我用的torch==2.6.0与torchvision==0.20.0冲突了，所有我下载了torch==2.5.0，然后cuda版本是12.4，我下载了flash_attn-2.7.4.post1+cu12torch2.5cxx11abiTRUE-cp310-cp310-linux_x86_64.whl解析了，出现了ImportError: /ssd/luying/anaconda3/envs/long-vita-1/lib/python3.10/site-packages/flash_attn_2_cuda.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN3c105ErrorC2ENS_14SourceLocationENSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE这个错误，如何解决

[shenyunhang]

可以试试我们的镜像：
shenyunhang/pytorch:24.11-py3_2024-1224
https://hub.docker.com/repository/docker/shenyunhang/pytorch/tags

或者nvidia提供的镜像：
https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch/tags

[421212-huv]

好的，感谢回复