Win10安装pytorch版本的alphapose,遇到了很多问题,经历了更换镜像源,CUDA重装等问题,经过一天半的调试,终于运行出来了。
CUDA版本:11.2
cudnn版本: 8.2.1
pytorch版本:1.7.1+cu110
python版本;3.6.5
1.1 创建虚拟环境(前提已经配置好anaconda)
在系统命令行cmd中输入
conda create -n Alphapose python=3.6
激活虚拟环境:
conda activate Alphapose
1.2 安装CUDA和CUDNN(较老的显卡)
(alphapose) C:\Users\GYG>conda install cudatoolkit=10.1
(alphapose) C:\Users\GYG>conda install cudnn=7.6.5
我的显卡是3060桌面版,如果是较新的显卡,就用不了。我是直接安装在本地的,网上有很多安装教程,这里就不赘述了。
参考链接:windows下cuda+cudnn安装
1.3 git 克隆alphapose代码
https://github.com/Amanbhandula/AlphaPose
打开pycharm终端,将项目下载到本地:
(alphapose) E:\projects>git clone https://github.com/Amanbhandula/AlphaPose
1.4 安装pytorch和torchvision,选择python解释器
(alphapose) E:\projects>conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=11.0 -c pytorch
选择Setting——>Python Interpreter——>Conda Environment——>Existing Environment,选择虚拟环境的python解释器;
1.5 模型文件下载
需要下载两个模型duc_se.pth和Yolov3-spp.weights分别放入models/sppe文件夹和models/yolo文件夹中,权重文件的百度云链接在项目主页如下位置:
1.6 依赖项安装
在terminal中安装requirements.txt中所要求的依赖项,其中红色框中依赖可省略。使用命令:
pip install *
至此环境配置部分已完成。
2.1 摄像头测试
- 复制readme.md文档中如下命令,
- 选择webcam_demo中的Edit Configurations
- 将命令粘贴到parameter参数列,–sp需要加上,点击apply,选择ok
4.接下来点击run按钮,运行
中间遇到了一些问题,记录如下
3.1 pycharm的终端下无法进入虚拟环境
解决方法如下:
- 打开pycharm,点击右上角齿轮,选择Setings
- 在tools下选择terminal,选择cmd.exe作为命令提示符,点击apply,ok
- 此时即可进入虚拟环境:
3.2 Argument ‘alpha’ can not be treated as a double
这是在fn.py文件中报的一个错误,有人说使用float()将变量改为单精度即可,但仍然没有解决问题。最后用的是github上面的方法:将opencv-python版本降低至3点几然后重新安装ffmpeg:
参考:https://www.bilibili.com/video/BV1hb4y117mu/
本文含有隐藏内容,请 开通VIP 后查看