DeepLearningExamples
DeepLearningExamples copied to clipboard
[MaskRCNN/PyTorch] PyTorch 1.11 breaks MaskRCNN/PyTorch benchmark build
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:
- Install PyTorch 1.11.0
- Checkout https://github.com/NVIDIA/DeepLearningExamples.git
-
bash DeepLearningExamples/PyTorch/Segmentation/MaskRCNN/requirements.sh
Expected behavior MaskRCNN/PyTorch benchmark builds successfully.
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