工作需要,配了一天环境,真的够够的,写篇文章,希望可以帮到更多的人,以及以后再配置环境也能通过这篇文章回想起来,如果大家有什么其他问题,可以在评论区提问,我知道的都会回答。
第一步:创建conda虚拟环境
conda create --name mmaction python=3.8 -y
conda activate mmaction
这里切记python版本不要超过3.9,有好多包,不支持更高版本python
第二步:创建安装pytorch
pytorch官网
pytorch文件有点大,下载较慢,可以使用下面的代码下载(优先官网):
pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple/ some-package --trusted-host mirrors.aliyun.com
通过在终端输入nvidia-smi查看自己的cuda版本,在pytorch官网找到与你系统和cuda版本对应的pytorch,进行安装。这里有个坑,cuda版本一定要和pytorch版本相同不然后面有的包安不上。
第三步:下载mmaction2源文件
可以使用git下载
git clone https://github.com/open-mmlab/mmaction2.git
也可以去官网下载
mmaction2-github
安装完后在终端进入mmaction2文件夹
第四步:安装包
pip install mmcv
pip install -r requirements/build.txt
安装最新版mmcv>2.0.0
第五步:运行测试
python demo/demo_inferencer.py demo/demo.mp4 --rec tsn --print-result --label-file tools/data/kinetics/label_map_k400.txt
如果pytorch是gpu版本会弹出视频后台也会进行推理,如果pytorch是cpu版本就只有后台推理,见下图
其他问题
如果出现ModuleNotFoundError: No module named 'mmaction’问题
在所运行的文件最上面添加如下代码:
import sys
sys.path.append('xxx\mmaction2')
xxx为你的绝对路径