Alphapose_pytorch版本环境配置Win10

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解释器;Alphapose_pytorch版本环境配置Win10_第1张图片
Alphapose_pytorch版本环境配置Win10_第2张图片
1.5 模型文件下载
需要下载两个模型duc_se.pth和Yolov3-spp.weights分别放入models/sppe文件夹和models/yolo文件夹中,权重文件的百度云链接在项目主页如下位置:Alphapose_pytorch版本环境配置Win10_第3张图片
1.6 依赖项安装
在terminal中安装requirements.txt中所要求的依赖项,其中红色框中依赖可省略。使用命令:

pip install *

Alphapose_pytorch版本环境配置Win10_第4张图片
至此环境配置部分已完成。

2.1 摄像头测试

  1. 复制readme.md文档中如下命令,Alphapose_pytorch版本环境配置Win10_第5张图片
  2. 选择webcam_demo中的Edit ConfigurationsAlphapose_pytorch版本环境配置Win10_第6张图片
  3. 将命令粘贴到parameter参数列,–sp需要加上,点击apply,选择okAlphapose_pytorch版本环境配置Win10_第7张图片
    4.接下来点击run按钮,运行在这里插入图片描述

中间遇到了一些问题,记录如下
3.1 pycharm的终端下无法进入虚拟环境
解决方法如下:

  1. 打开pycharm,点击右上角齿轮,选择SetingsAlphapose_pytorch版本环境配置Win10_第8张图片
  2. 在tools下选择terminal,选择cmd.exe作为命令提示符,点击apply,okAlphapose_pytorch版本环境配置Win10_第9张图片
  3. 此时即可进入虚拟环境:Alphapose_pytorch版本环境配置Win10_第10张图片
    3.2 Argument ‘alpha’ can not be treated as a double
    这是在fn.py文件中报的一个错误,有人说使用float()将变量改为单精度即可,但仍然没有解决问题。最后用的是github上面的方法:将opencv-python版本降低至3点几然后重新安装ffmpeg:Alphapose_pytorch版本环境配置Win10_第11张图片
    参考:https://www.bilibili.com/video/BV1hb4y117mu/

你可能感兴趣的:(pytorch,python,深度学习)