第一步:从官方网站下载 Miniconda,链接Miniconda — miniconda documentation
第二步:创建一个 conda 环境并激活它
conda create --name openmmlab(可以修改) python=3.8(根据你的python版本) -y
激活环境
conda activate openmmlab
第三步:安装 PyTorch,按照官方说明进行操作,例如
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
此处要参考自己的cuda版本,网上有很多例子
检验是否安装好,命令行输入python
然后运行
import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用 TRUE
print(torch.cuda.device_count()) # 查看可用的CUDA数量 1
print(torch.version.cuda) # 查看CUDA的版本号 版本
结果为后面的证明安装成功
第四步:使用 MIM 安装 MMEngine、MMCV、MMDetection(可选)和 MMPose(可选)。后俩个一般做检测和姿态估计。
pip install -U openmim
mim install mmengine
mim install mmcv
mim install mmdet
mim install mmpose
安装mmcv有问题可以参考安装MMCV血泪史_mmcv源码安装-CSDN博客
第五步:从源代码安装 mmaction2(建议)git clone不下来的,可以私聊我要压缩文件,解压后,进入目录执行后俩个命令即可。
git clone https://github.com/open-mmlab/mmaction2.git
cd mmaction2
pip install -v -e .