Alphapose_pytorch版本环境配置Win10

Thuggish

已于 2022-09-05 22:42:36 修改

阅读量1.1k

点赞数 1

文章标签： pytorch python 深度学习

于 2022-09-05 22:38:17 首次发布

本文链接：https://blog.csdn.net/qq_43547141/article/details/126713835

版权

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/