安装mmaction环境

环境配置

1.首先创建conda环境

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

通过conda info --env可以查看 

安装Pytorch 

2.安装pytorch,此处需要根据官网和服务器配置选择对应的版本,我的cuda是10.2,安装的版本为1.12.0.PyTorch

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

安装MIM

1.安装mmcv-full

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10.0/index.html

2.克隆MMAction2库

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

3.安装依赖包和MMAction2

pip install -r requirements/build.txt
pip install -v -e .  # or "python setup.py develop"
#我这里使用的是python setup.py develop

 安装好了之后就进行测试,这里我是手动重新装了一下mmcv-full,因为版本跟我的torch=1.12.0不匹配。测试代码如下:

安装mmaction环境_第1张图片

安装依赖

pip install -r requirements/build.txt
python setup.py develop
pip install decord -i https://pypi.douban.com/simple

这里备注一下我踩的坑,我训练slowfast的时候使用tool/train.py报错说找不到gpu_ids:

报错“ConfigDict ”object has no attribute “gpu_ids”

我当时用的应该是mmcv-full1.6.0和torch1.12.0版本,后来发现是版本不兼容吧,我将torch改成了1.11.0和mmcv-full=1.4.7之后解决了.(cuda=10.2没改过) 

 

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