mmskeleton安装

最近开始尝试ST-GCN，主要安装步骤参考作者在github上提供的指南：https://github.com/open-mmlab/mmskeleton/blob/master/doc/GETTING_STARTED.md
本文记录一下安装过程

安装

安装Anaconda3

安装cuda、cudnn、nccl

版本：

cuda-10.1
cudnn-7.6（应该是当时对应cuda10.1的最新版cudnn）
nccl-2.6（也是对应cuda10.1的最新版nccl）

新建conda环境并激活

新建一个名为open-mmlab，python版本为3.7的conda虚拟环境（也可以不需要新建）。

conda create -n open-mmlab python=3.7 
conda activate open-mmlab

安装与cuda版本对应的pytorch

这部分参考pytorch官网的指南。

记得添加国内镜像地址到安装源，否则非常慢。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

将mmskeleton从github下载到本地

git clone https://github.com/open-mmlab/mmskeleton.git
cd mmskeleton

安装mmskeleton

python setup.py develop

注