search

deep-dance / core

GNU General Public License v3.0
2 stars 1 forks source link

Running detectron2 fails with ROCm #8

Open erak opened 4 years ago

erak commented 4 years ago
  1. Use pytorch docker image built for ROCm as described here: https://rocmdocs.amd.com/en/latest/Deep_learning/Deep-learning.html#recommended-install-using-published-pytorch-rocm-docker-image
  2. Then install OpenCV and missing dependencies, create new docker image using docker commit
  3. Build detectron2, run on example.
  4. Inference fails because torchvision is not built with CPU support (building it with OpenCL / ROCm WITH_HIP fails because of some weird compilation error in caffe2)