近期在学习slowfast,在配置环境的时候踩了很多坑,以此记录一下,也方便各位同学踩坑的时候参考一下:
第一步:conda创建新环境
conda create -n SlowFast python=3.7
这里我用的是python3.7,根据需要这里可以替换。
第二步:安装pytorch
pip3 install torch==1.10.0+cu102 torchvision==0.11.1+cu102 torchaudio===0.10.0+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html
注意:这里我在后续安装遇到了一些问题,需要保证cuda版本大于10.2,10.1遇到了各种本人能力无法解决的问题。
第三步:下载SlowFast源码
https://github.com/facebookresearch/SlowFast
从GitHub上面下载官方源码,这里可能需要一些小操作,看个人工具了。
第四步:安装cocoapi社区版
下载以下链接的cocoapi,因为cocoapi官方暂不支持Windows
https://github.com/philferriere/cocoapi
第五步:下载fvcore
下载链接:
https://github.com/facebookresearch/fvcore
第三、四、五步均来自以下文章:
————————————————
版权声明:本文为CSDN博主「iamleifeng」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/iamleifeng/article/details/103877581
第六步:下载detectron2源码
下载地址:
https://github.com/conansherry/detectron2
第七步:安装detectron2
到此为止,我们可以得到一个文件夹,如下所示:
先安装cocoapi-master
cd cocoapi-master/PythonAPI
python setup.py build_ext --inplace
python setup.py build_ext install
安装fvcore
cd fvcore-main
python setup.py build --force develop
到此为止,还不能够安装detectron2,虽然我的anaconda环境里面已经安装cuda和cudnn了,但是编译的时候,需要电脑环境中也有,这一点我也不知道为什么,
cuda 10.2 下载地址
https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
cudnn下载地址
https://developer.nvidia.com/rdp/cudnn-download
安装方式参考如下:
https://blog.csdn.net/sinat_23619409/article/details/84202651
到此,还没完,需要自行安装Visual Studio,我自己用的是VS2019。
修改如下文件
detectron2\detectron2\layers\csrc\nms_rotated\nms_rotated_cuda.cu 中第11行新增一行。(不同版本行数不一样)
1 #define WITH_HIP// 添加这一行
2 #ifdef WITH_HIP
到此,我在我电脑上的环境部署已经完成,进行detectron2的编译:
cd detectron2
python setup.py build --force develop
你以为能顺利结束了么,不可能的!!!
在编译过程中,会时不时的卡在一个位置,那就是你缺少所需要的安装包,提示你缺啥,用
pip install去安装一下 就好了。
到此 deteceron2的安装完毕。下一步 进行编译slowfast。
第八步:编译SlowFast
打开SlowFast源码,找到steup.py文件,修改如下:
PIL库被pillow代替,其他所需要的库进行手动安装,基本上pip install都可以直接安装。
到此,大功告成!!!
SlowFast真的需要需要需要好多好多好多东西东西东西,附上一段我自己的测试视频的截图,效果不好,只是为了测试一下是否好用。
祝各位:研途愉快!