Почему USE_OPENCV выключен при установке PyTorch? - PullRequest
0 голосов
/ 20 сентября 2018

В CMakeLists.txt в папке pytorch используйте Opencv, ВКЛЮЧЕНО.

option(USE_OPENCL "Use OpenCL" OFF)
option(USE_OPENCV "Use OpenCV" ON)
option(USE_OPENMP "Use OpenMP for parallel code" OFF)

В setup.py переменная установлена ​​с Opencv как

use_env_vars = ['CUDA', 'CUDNN', 'MIOPEN', 'MKLDNN', 'NNPACK', 'DISTRIBUTED',
                'OPENCV', 'SYSTEM_NCCL', 'GLOO_IBVERBS']

На моем ПК установлен Opencvи может проверить с помощью pkg-config --modversion opencv дает мне 3.3.0.

OpenCV установлен на моем компьютере.Но во время установки Caffe2 он показал мне

--   USE_NERVANA_GPU       : OFF
--   USE_NNPACK            : 1
--   USE_OBSERVERS         : ON
--   USE_OPENCL            : OFF
--   USE_OPENCV            : OFF
--   USE_OPENMP            : OFF
--   USE_PROF              : OFF
--   USE_REDIS             : OFF
--   USE_ROCKSDB           : OFF
--   USE_ZMQ               : OFF

OpenCV выключен.Как я могу включить OpenCV в Caffe2?

1 Ответ

0 голосов
/ 28 ноября 2018

Вы можете использовать что-то вроде этого:

USE_OPENCV = 1 USE_FFMPEG = 1 USE_LMDB = 1 python setup.py install

...