cuda9.0 instalar mmdetection cuda y problema de falta de coincidencia de pytorch

cuda no coincide con mmdet
El controlador cuda9.0 instala mmdetection, y aparecerá la instalación de acuerdo con los pasos del sitio web oficial. El problema de que cuda no puede coincidir con mmdetection, de hecho, esta vez es la razón por la cual la versión de pytorch no coincide con la versión del controlador cuda. Debido a que sólo es compatible con cuda9.0 pytorch1.1, el apoyo a torchvision 0.3.0.
Así que necesitamos
pip uninstall torch

pip uninstall torchvision
Luego instale la versión especificada y
conda install -n open-mmlab pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
luego
pip install -v -e .

Publicado 6 artículos originales · recibido 1 · vistas 468

Supongo que te gusta

Origin blog.csdn.net/weixin_43590151/article/details/104642021
Recomendado
Clasificación