cuda9.0ドライバーがmmdetectionをインストールし、公式サイトの手順に沿ったインストールが表示されますが、実際にはpytorchのバージョンがcudaドライバーのバージョンと一致しないのは、このため、mmudadetectionと一致しません。cuda9.0は0.3.0にpytorch1.1、torchvisionのサポートをサポートしているため。
私たちが必要とするので
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 .