DeepLearningExamples icon indicating copy to clipboard operation
DeepLearningExamples copied to clipboard

[MaskRCNN/PyTorch] PyTorch 1.11 breaks MaskRCNN/PyTorch benchmark build

Open ghost opened this issue 2 years ago • 1 comments

Related to MaskRCNN/PyTorch 1.11

Describe the bug MaskRCNN/PyTorch benchmark failed to build with PyTorch 1.11. PyTorch 1.11 removes most of the THC headers with https://github.com/pytorch/pytorch/pull/69929. This breaks the benchmark build since there are dependencies within DeepLearningExamples/PyTorch/Segmentation/MaskRCNN/pytorch/maskrcnn_benchmark/csrc/cuda/*.

To Reproduce Steps to reproduce the behavior:

  1. Install PyTorch 1.11.0
  2. Checkout https://github.com/NVIDIA/DeepLearningExamples.git
  3. bash DeepLearningExamples/PyTorch/Segmentation/MaskRCNN/requirements.sh

Expected behavior MaskRCNN/PyTorch benchmark builds successfully.

ghost avatar Mar 11 '22 18:03 ghost

I tried similar fix as https://github.com/NVIDIA/DeepLearningExamples/commit/bd4cd216192ed43831cd66a58dd7ea79a39bcac3. It works for PyTorch 1.11 but it will break build < PyTorch 1.11.0

ghost avatar Mar 11 '22 18:03 ghost