Bench2Drive github:https://github.com/Thinklab-SJTU/Bench2Drive
Bench2DriveZoo github:https://github.com/Thinklab-SJTU/Bench2DriveZoo
在复现Bench2Drive的时候遇到的问题,显示
FutureWarning: `torch.cuda.amp.custom_fwd(args...)` is deprecated. Please use `torch.amp.custom_fwd(args..., device_type='cuda')` instead.
当前环境中安装的torch版本是2.4.0,找了半天原因,终于在github的一个issue下面找到了原因。
原因:
torch从2.4及以后便不再支持`torch.cuda.amp.custom_fwd(args...)`
解决方法:
将torch版本降级为2.3,torchvision版本也需要相应降低。
版本对应参考这里:https://github.com/pytorch/vision#installation
我用的版本是这样的:
pip install torch==2.3.1+cu118 torchvision==0.18.1+cu118 torchaudio -f https://download.pytorch.org/whl/torch_stable.html
这之后问题及基本就解决了。
补充:
建议在配置环境的时候就早些修改,如果安装完了建议把一些模块比如mmcv--0.01卸了再重新“pip install -v -e.”。否则,有些依赖改起来很费劲。