2. Установите Питорч

     conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

или:

pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

3.Установить пикокотулс

pip install git+GitHub.com/Фил Фил Дэй и…

4. Установите mmcv

pip install mmcv

5. Установите ммдетект

Официальный сайт:GitHub.com/open-MM lab/…

git clone GitHub.com/open-MM lab/…

cd mmdetection

pip install -r requirements.txt

Измените скомпилированный файл:

.Исправьте info = info.decode().lower() в ~Lib\site-packages\torch\utils\cpp_extension.py до info=info.decode("utf8","ignore").lower()

Все decode() изменены на decode("utf8","ignore")

Измените файл setup.py для mmdetection-master.

будет:

def make_cuda_ext(name, module, sources, sources_cuda=[]):

define_macros = []

extra_compile_args = {'cxx': []}

if torch.cuda.is_available() or os.getenv('FORCE_CUDA', '0') == '1':

define_macros += [('WITH_CUDA', None)]

extension = CUDAExtension

extra_compile_args['nvcc'] = [

'-D__CUDA_NO_HALF_OPERATORS__',

'-D__CUDA_NO_HALF_CONVERSIONS__',

'-D__CUDA_NO_HALF2_OPERATORS__',

]

sources += sources_cuda

else:

print(f'Compiling {name} without CUDA')

extension = CppExtension

# raise EnvironmentError('CUDA is required to compile MMDetection!')

return extension(

name=f'{module}.{name}',

sources=[os.path.join(*module.split('.'), p) for p in sources],

define_macros=define_macros,

extra_compile_args=extra_compile_args)

изменить на:

def make_cuda_ext(name, module, sources, sources_cuda=[]):

return CUDAExtension(
name='{}.{}'.format(module, name),
sources=[os.path.join(*module.split('.'), p) for p in sources],
extra_compile_args={
'cxx': ["-DMS_WIN64","-MD"],
'nvcc': [
'-D__CUDA_NO_HALF_OPERATORS__',
'-D__CUDA_NO_HALF_CONVERSIONS__',
'-D__CUDA_NO_HALF2_OPERATORS__',
]
})

Затем запустите:

python setup.py build_ext --inplace для компиляции

python setup.py install develop для завершения установки

После завершения выполнения запустите pip list для просмотра:

Добавлено одно предложение: Даже после успешной установки его нельзя использовать! Новая версия больше не поддерживает Windows! TMD, это то, что люди делают