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、在相关的上下载，解压缩，进到文件夹里，运行

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