首先要确保自己的环境符合要求,参考地址
一、CUDA
博主的笔记本GPU是GTX 1050TI,驱动最高支持版本到CUDA11.5,这里博主还是选用了CUDA10.1,考虑到各种兼容问题。
CUDA的安装方法可以自行搜查,这里不再赘述。
安装完成后,使用nvcc -V查看cuda是否安装成功。
二、Pytorch、Torch
安装对应CUDA版本的pytorch,在这里找到对应的安装命令,使用命令行安装即可。
三、Visual Studio
为了在Windows上安装mmcv,需要VS里的MSVC工具cl.exe,建议安装Visual Studio Community 2019(2017也可以)。
博主亲测VS2022由于MSVC工具版本太新无法顺利编译。
四、mmcv-full
可以参考文章最后的文章步骤,这里不再赘述。此外,注意mmcv的版本问题(点击链接可以查看),要使用GPU,CUDA版本一定要有支持的mmcv版本。
遇到的问题:
编译报错:error: member “torch::jit::ProfileOptionalOp::Kind“ may not be initialized
解决方案:https://blog.csdn.net/qq_40714949/article/details/115654215
编译报错:OSError: symbolic link privilege not held
解决方案:以管理员身份运行命令行or Conda Prompt
五、mmaction2、mmdetection
git clone https://github.com/open-mmlab/mmaction2.git
cd mmaction2
pip install -r requirements/build.txt
pip install -v -e . # or "python setup.py develop"
pip install mmdet
如果以上命令安装不成功,可以参考以下的官方文档,有更详细的说明:
mmaciton2安装指南
mmdetction安装指南
六、安装参考
mmcv-full安装报错
Windows下mmcv-full1.3.13安装指南
OpenMMlab官方——Windows下mmcv-full安装指南(参考同时编译CPU和GPU的OPs)