AlphaPose环境配置及试运行(2023.6)

本文介绍了在Windows子系统Linux(WSL2)上的Ubuntu20.04环境中,如何配置CUDA11.3、PyTorch1.12.1和Python3.7的详细过程,包括conda环境创建、CUDA的下载安装、Python的版本匹配以及PyTorch的pip安装。此外,还提到了AlphaPose项目的克隆与安装,以及安装过程中可能遇到的包依赖和环境变量配置问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

环境配置

我的环境:WSL2(Ubuntu20.04+cuda11.3+pytorch1.12.1+Python3.7)
根据官方Git上提供的INSTALL.RM一步一步安装(但是会遇到很多坑)

创建conda环境

conda create -n alphapose python=3.7 -y
conda activate alphapose

安装cuda

wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run
sudo sh cuda_11.3.0_465.19.01_linux.run

不同版本详询NVIDIA官网
注意cuda版本需与显卡算力匹配(我就是到最后一步了才发现不匹配QAQ)
匹配关系见这篇文章
安装完需要配置环境变量

export PATH=/usr/local/cuda-11.3/bin/:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64/:$LD_LIBRARY_PATH

路径要改成自己安装的版本路径
注意:此方法是暂时性的,每次打开linux需重新配置。
可用nvcc -V命令查看安装的cuda版本

安装Python

官方的pytorch安装方式如下,但是我跑不起来

conda install pytorch torchvision torchaudio pytorch-cuda=11.3 -c pytorch -c nvidia 

自行安装(匹配cuda版本),pytorch官网查看安装方式,建议使用pip安装,不要用conda,conda有可能安装cup版本,以下是我的安装方式:

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

验证安装是否成功,进入Python执行以下语句

import torch
print(torch.__version__) #查看pytorch版本
print(torch.cuda.is_available()) #返回True说明版本是匹配的

下载alphapose

git clone https://github.com/MVIG-SJTU/AlphaPose.git
cd AlphaPose

安装其他包

python -m pip install cython
sudo apt-get install libyaml-dev
################Only For Ubuntu 18.04#################
locale-gen C.UTF-8
# if locale-gen not found
sudo apt-get install locales
export LANG=C.UTF-8
######################################################
python setup.py build develop

有的包安装不上可以自行使用pip安装,然后重跑setup

出现以下输出代表环境安装成功,cython-bbox安装不成功的话可以多试几次,应该是网络原因

DEMO试运行

参考这篇文章
可能出现的问题:
报错NotImplementedError(),且模型无法成功加载
环境变量重新配置一下,因为前面配置的方式是临时的

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值