TransT环境配置问题

1.个人平台说明

ubuntu18.04,RXT3060,pycharm

2.环境配置记录

按照README.md文件依次进行配置(包括激活命名为transt的conda环境),配置完成之后,在工程主目录TransT下面创建一个networks文件夹,然后将下载的预训练模型transt.pth移动到里面。然后修改run_video.py中的参数,然后运行run_video.py。出现以下错误,按顺序进行记录。

2.1.报错“RuntimeError:CUDA error:no kernel image is available for execution on the device

解决方案:
更新pytorch、torchvision的版本。我个人从代码作者推荐的pytorch=1.5 torchvision=0.6.1 cudatoolkit=10.2更新到了torch1.9.1+cu111 torchvision0.10.1+cu111 torchaudio==0.9.1。具体命令如下:

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
想下载其它版本请参考网址:https://pytorch.org/get-started/previous-versions/

2.2.报错“ImportError: cannot import name ‘int_classes’ from ‘torch._six’”

解决方案:
将from torch._six import string_classes, int_classes注释掉,然后添加int_classes = int、string_classes = str,查阅得知原因可能是torch版本和torchvision版本不对应,如果是版本不对应就要换版本了。如果不是因为版本原因报错的,这样修改有效,具体如下:

# from torch._six import string_classes, int_classes
int_classes = int
string_classes = str

2.3.报错“cannot import name ‘_new_empty_tensor’ from ‘torchvision.ops’”

解决方案:
将下面三行代码直接注释掉。可能的原因请参照:http://www.taodudu.cc/news/show-5758046.html?action=onClick

# if float(torchvision.__version__[:3]) < 0.7:
#     from torchvision.ops import _new_empty_tensor
#     from torchvision.ops.misc import _output_size

3.参考

  1. MixFormer目标跟踪CVPR2022(代码调试):http://www.taodudu.cc/news/show-5758046.html?action=onClick
  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值