cuda9.0安装mmdetection的cuda与pytorch不匹配问题

cuda9.0的驱动安装mmdetection，按照官网的步骤安装会出现，cuda不能与mmdetection匹配的问题，其实这个时候是pytorch的版本与cuda驱动版本不匹配的原因。因为cuda9.0只支持到pytorch1.1，torchvision支持到0.3.0.
故我们需要
pip uninstall torch

pip uninstall torchvision
然后安装指定版本
conda install -n open-mmlab pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
然后
pip install -v -e .