ModuleNotFoundError: No module named 'fast_layer_norm'

[Zhefei777]

❓ Questions and Help

Hi there, I have a problem. I have installed apex and maskrcnn_benchmark successfully. But when I ran demo.py , I got this.
File "webcam.py", line 6, in
from predictor import COCODemo
File "/home/zhefei/maskrcnn-benchmark/demo/predictor.py", line 6, in
from maskrcnn_benchmark.modeling.detector import build_detection_model
File "/home/zhefei/maskrcnn-benchmark/maskrcnn_benchmark/modeling/detector/init.py", line 2, in
from .detectors import build_detection_model
File "/home/zhefei/maskrcnn-benchmark/maskrcnn_benchmark/modeling/detector/detectors.py", line 2, in
from .generalized_rcnn import GeneralizedRCNN
File "/home/zhefei/maskrcnn-benchmark/maskrcnn_benchmark/modeling/detector/generalized_rcnn.py", line 11, in
from ..backbone import build_backbone
File "/home/zhefei/maskrcnn-benchmark/maskrcnn_benchmark/modeling/backbone/init.py", line 2, in
from .backbone import build_backbone
File "/home/zhefei/maskrcnn-benchmark/maskrcnn_benchmark/modeling/backbone/backbone.py", line 7, in
from maskrcnn_benchmark.modeling.make_layers import conv_with_kaiming_uniform
File "/home/zhefei/maskrcnn-benchmark/maskrcnn_benchmark/modeling/make_layers.py", line 10, in
from maskrcnn_benchmark.layers import Conv2d
File "/home/zhefei/maskrcnn-benchmark/maskrcnn_benchmark/layers/init.py", line 10, in
from .nms import nms
File "/home/zhefei/maskrcnn-benchmark/maskrcnn_benchmark/layers/nms.py", line 5, in
from apex import amp
File "/home/zhefei/.conda/envs/maskrcnn_benchmark/lib/python3.8/site-packages/apex/init.py", line 23, in
from . import transformer
File "/home/zhefei/.conda/envs/maskrcnn_benchmark/lib/python3.8/site-packages/apex/transformer/init.py", line 4, in
from apex.transformer import pipeline_parallel
File "/home/zhefei/.conda/envs/maskrcnn_benchmark/lib/python3.8/site-packages/apex/transformer/pipeline_parallel/init.py", line 1, in
from apex.transformer.pipeline_parallel.schedules import get_forward_backward_func
File "/home/zhefei/.conda/envs/maskrcnn_benchmark/lib/python3.8/site-packages/apex/transformer/pipeline_parallel/schedules/init.py", line 5, in
from apex.transformer.pipeline_parallel.schedules.fwd_bwd_no_pipelining import forward_backward_no_pipelining
File "/home/zhefei/.conda/envs/maskrcnn_benchmark/lib/python3.8/site-packages/apex/transformer/pipeline_parallel/schedules/fwd_bwd_no_pipelining.py", line 10, in
from apex.transformer.pipeline_parallel.schedules.common import Batch
File "/home/zhefei/.conda/envs/maskrcnn_benchmark/lib/python3.8/site-packages/apex/transformer/pipeline_parallel/schedules/common.py", line 6, in
from apex.contrib.layer_norm.layer_norm import FastLayerNorm
File "/home/zhefei/.conda/envs/maskrcnn_benchmark/lib/python3.8/site-packages/apex/contrib/layer_norm/init.py", line 1, in
from .layer_norm import FastLayerNorm
File "/home/zhefei/.conda/envs/maskrcnn_benchmark/lib/python3.8/site-packages/apex/contrib/layer_norm/layer_norm.py", line 5, in
import fast_layer_norm
ModuleNotFoundError: No module named 'fast_layer_norm'

I use torch 1.6+ torchvision 0.7.0 and cuda 10.1
Thanks in advance!!