Areium的博客

原创 Swin-Transform Windows环境搭建踩坑记录

这里安装mmcv-full 1.5.0版本的库，这很重要。在我安装尝试中，新版的mmcv库模型权重文件和库里的参数对不上，即使能运行，跑出来的结果也是全零。而在库里给出的mmcv的版本为1.2.4~1.4.0，在多次安装尝试中，发现这个区间的版本在Windows环境下可能会不支持cuda里的nvcc，显示nvcc not available。我用的是官方库的Swin-Transformer-Object-Detection，该库的实现用的是比较旧版的环境，因此选择1.X的torch环境。