关于VMamba下,No module named ‘torch._six‘的解决方案(通用其他模型)
mamba
项目地址:https://gitcode.com/gh_mirrors/ma/mamba
免费下载资源
·
一、前言:
对于Vmamba而言,配置环境是十分脆弱的。那么当我们遇到版本兼容性问题是,首先想到的解决方案是去修改代码,而并不是修改依赖库。
二、报错内容:
我们发现如下报错内容:
File "/home/VMamba/classification/utils/utils.py", line 11, in <module>
from torch._six import inf
ModuleNotFoundError: No module named 'torch._six'
ERROR:torch.distributed.elastic.multiprocessing.api:failed (exitcode: 1) local_rank: 0 (pid: 17457) of binary: /home/username/anaconda3/envs/vim/bin/python
此原因是因为pytorch2.0已经去除了torch._six这一用法。按照前文描述,我们在不希望区更改pytorch版本的情况下来解决此问题。
那么此时我们就需要按照报错内容修改代码。
三:修改:
##我们将
torch._six
##修改为
torch
即可解决问题。
GitHub 加速计划 / ma / mamba
12.02 K
1.01 K
下载
最近提交(Master分支:3 个月前 )
62db608d
3 个月前
323db269
Co-authored-by: Tri Dao <tridao@users.noreply.github.com> 3 个月前
更多推荐
已为社区贡献2条内容
所有评论(0)