search

Human9000 / nd-Mamba2-torch

Only implemented through torch: "bi - mamba2" , "vision- mamba2 -torch". support 1d/2d/3d/nd and support export by jit.script/onnx;
191 stars 7 forks source link

请问能在pytorch 的版本要求是多少？能在1.12.0上面运行吗？ #8

Open Fatima-Ito opened 2 months ago

Fatima-Ito commented 2 months ago

谢谢🙏

Human9000 commented 2 months ago

感谢你的提问，下面是回答： 本项目使用的环境是torch2，具体的版本没有很强的依赖性 针对torch1的问题，你可以自行测试，应该可以直接运行。 如果不行的话，可能是某些函数结构不完全一致，简单的修改应该可以使用。 如果你仍然无法完成的话，请留言，后续我可以尝试修改成1.12.0的适用版本，单独发给你

Fatima-Ito commented 2 months ago

感谢你的提问，下面是回答： 本项目使用的环境是torch2，具体的版本没有很强的依赖性 针对torch1的问题，你可以自行测试，应该可以直接运行。 如果不行的话，可能是某些函数结构不完全一致，简单的修改应该可以使用。 如果你仍然无法完成的话，请留言，后续我可以尝试修改成1.12.0的适用版本，单独发给你

好的，非常感谢您🙏🙏🙏🙏

CCCCHENnn commented 2 months ago

我的pytorch为1.12也使用不了，求发 @Human9000

Human9000 commented 1 month ago

经测试，该项目在如下的环境可以使用，请检查你本地的环境是否存在问题。 cudatoolkit 11.6.0 hecad31d_10 conda-forge pytorch 1.12.1 py3.9_cuda11.6_cudnn8.3.2_0 pytorch