Flownet 2.0-pytorch window10编译过程

github上NVIDIA给出的flownet2-pytorch代码

我使用的环境

python 3.5
pytorch 1.0.1
cuda 9.0
cudnn 7.1
这里最重要的是你的cuda和pytorch版本要相对应，否则会有好多坑，建议用conda装。
除此之外，还有Microsoft Visual C++ Build Tools 14。

编译过程

1. 分别进入networks下的这三个文件夹，然后找到setup.py,
   
   在setup.py中呢，需要将cxx_args = ['-std=c++11']中的11改成14，因为前面我是下的VC 14，这个根据自己的情况来。
   然后在nvcc_args这个参数中，'-gencode', 'arch=compute_35,code=sm_35'，这个里面的数字是对应你的显卡的算力，这个大家根据自己的显卡然后修改，我的算力是3.5，所以改成了35。
   至此，setup.py就修改完了，三个文件夹里的都要修改。
   修改完之后，分别在终端中进入这三个文件夹，然后输入python setup.py install这个命令，
   如果报下面这个错误的话，

invalid preprocessor command 'warning'

就得找到你的torch.h文件，我的路径是：

E:\Miniconda\install\envs\flow\Lib\site-packages\torch\lib\include\torch\csrc\api\include\torch\torch.h

找到t