pwc-net 的pytorch运行

， cuda版本比较新，我是1.6.0和10.0
2）
这个是备选，但是没用上，不太清楚。
3）这是官网，如果你是.04，可以在评论区里找到，自己安装太费劲了，要求cuda8，.2.0，.7
安装的过程
1、创建虚拟环境

文章插图
2、安装cupy（这里就要手动下载，也许你可以更改的源试试看，我是亚马逊的服务器，有时候还可以，所以不想改）
1）下载numpy
2）安装cupy ，使用以下命令，可惜网太慢，去这里手动下载吧
pip install cupy
安装好cupy的依赖，再安装一次cupy 。因为之前安装的时候依赖没下载下来，所以cupy根本没安装成功。
3、安装

文章插图
pip install torch==1.6.0 torchvision==0.7.0 -i https://pypi.doubanio.com/simple
安装过.8，会报错：: cuDNN error:
网上说是版本不匹配的问题。作者的要求的>=1.6.0，所以卸载

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

安装1.6.0版本就好用啦
4、在相关的上下载，解压缩，进到文件夹里，运行
【pwc-net 的pytorch运行】python run.py --model default --one ./images/one.png --two ./images/two.png --out ./out.flo