在配置flownet-pytorch环境的时候需要安装一个包——spatial-correlation-sampler

pip install spatial-correlation-sampler

会出现报错信息

ERROR:Command errored out with exit status 1:…

解决方法

export CUDA_HOME=/usr/local/cuda-10.0
pip install spatial-correlation-sampler

安装成功。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐