AI Studio 配置 mmdetection 环境

AI Studio 配置 mmdetection 环境

首先进入 AI Studio 的黑窗，按如下步骤操作（不要在 notebook 中操作）
黑窗



一、conda 换源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes


二、安装 pytorch

1、先使用 nvcc -V 查看 cuda 版本
cuda
可以看到 cuda 版本为 9.2


2、创建虚拟环境

conda create -n mmdet python=3.7
conda init

关闭黑窗并重新打开，并运行：

conda activate mmdet

此时你会进入mmdet的虚拟环境中。（如下图）
env


3、安装 pytorch

conda install pytorch torchvision cudatoolkit=9.2

cudatoolkit 版本根据 cuda 版本进行安装，因为 cuda 版本为 9.2，所以这里也安装 9.2 版本的 cudatoolkit。



三、安装 mmdetection

1、安装 mmcv

pip install mmcv

2、安装 mmdet

git clone https://github.com/open-mmlab/mmdetection
cd mmdetection
pip install -r requirements.txt
python setup.py develop

提示安装成功！
finish

我们再pip查看一下：

pip list

ok

安装成功了！！！

原创文章，如果配置好了请点个赞。谢谢！