I meet some questions No module named 'correlation_cuda'

[huangjiangui]

(base) root@autodl-container-c6d211963c-81d44250:/autodl-fs/FastFlowNet-main# python benchmark.py
Traceback (most recent call last):
File "benchmark.py", line 5, in
from models.FastFlowNet import FastFlowNet
File "/root/autodl-fs/FastFlowNet-main/models/FastFlowNet.py", line 5, in
from .correlation_package.correlation import Correlation
File "/root/autodl-fs/FastFlowNet-main/models/correlation_package/correlation.py", line 4, in
import correlation_cuda
ModuleNotFoundError: No module named 'correlation_cuda'
(base) root@autodl-container-c6d211963c-81d44250:/autodl-fs/FastFlowNet-main#

[ltkong218]

May be you should replace from models.FastFlowNet import FastFlowNet with from models.FastFlowNet_v2 import FastFlowNet in benchmark.py. Also, please refer to the README.md file to configure the correct environment.