环境配置
我的环境: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(),且模型无法成功加载
环境变量重新配置一下,因为前面配置的方式是临时的