pwc-net 的pytorch运行

遇到的问题

pip install太慢导致安装失败

1）使用conda install，事先得更改镜像源，网上搜conda镜像源一大堆
2）镜像源中找不到包，则复制pip install下给出的链接，fan qiang后粘贴到浏览器，可直接下载.whl文件；下载完毕，再使用pip install安装.whl文件即可。
3）在pip后面加上-i https://pypi.doubanio.com/simple
类似下面这个形式

pip install --target=/root/miniconda2/lib/python2.7/site-packages opencv-contrib-python==4.1.0.25 -i https://pypi.doubanio.com/simple

还有可能pip给的链接不全，只能自己搜官网下载，不过可以供参考，以免下错。

本机环境

ubuntu18.04

cuda10.0，cudnn7.5.1，cupy9.5.0，python3.7，numpy1.21.0，pillow8.4.0

可参考的github

1）收藏数463，也是本人参考https://github.com/sniklaus/pytorch-pwc
pytorch，cuda版本比较新，我是1.6.0和10.0
2）https://github.com/zyong812/pwc-net_Pytorch
这个是备选，但是没用上，不太清楚。
3）https://github.com/NVlabs/PWC-Net这是官网，如果你是Ubuntu16.04，可以在评论区里找到dockerfile，自己安装太费劲了，要求cuda8，pytorch0.2.0，python2.7

安装pwcnet的过程

1、创建虚拟环境
2、安装cupy（这里就要手动下载，也许你可以更改ubuntu的源试试看，我是亚马逊的服务器，有时候还可以，所以不想改）
1）下载numpy
2）安装cupy，使用以下命令，可惜网太慢，去这里手动下载吧

pip install cupy

安装好cupy的依赖，再安装一次cupy。因为之前安装的时候依赖没下载下来，所以cupy根本没安装成功。

3、安装pytorch

pip install torch==1.6.0 torchvision==0.7.0 -i https://pypi.doubanio.com/simple

安装过pytorch1.8，会报错：RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
网上说是版本不匹配的问题。作者的要求的pytorch>=1.6.0，所以卸载

pip uninstall torch
#可能还需要卸载下面这个，看你安装1.6.0版本会不会有错，报哪个冲突就删哪个
pip uninstall torchaudio==0.8.0

安装1.6.0版本就好用啦

4、在相关的github上下载pwcnet，解压缩，进到文件夹里，运行

python run.py --model default --one ./images/one.png --two ./images/two.png --out ./out.flo