一、前言:

对于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 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐