win10 下配置 slowfast 环境

近期在学习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真的需要需要需要好多好多好多东西东西东西,附上一段我自己的测试视频的截图,效果不好,只是为了测试一下是否好用。

祝各位:研途愉快! 

 

  • 4
    点赞
  • 36
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值