pwc
遇到的问题
pip install太慢导致安装失败
1)使用conda install,事先得更改镜像源,网上搜conda镜像源一大堆
2)镜像源中找不到包,则复制pip install下给出的链接,fan qiang后粘贴到浏览器,可直接下载.whl文件;下载完毕,再使用pip install安装.whl文件即可。
3)在pip后面加上-i
类似下面这个形式
pip install --target=/root/miniconda2/lib/python2.7/site-packages opencv-contrib-python==4.1.0.25 -i
还有可能pip给的链接不全,只能自己搜官网下载,不过可以供参考,以免下错。
本机环境
ubuntu18.04
cuda10.0,cudnn7.5.1,cupy9.5.0,python3.7,numpy1.21.0,pillow8.4.0
可参考的github
1)收藏数463,也是本人参考
pytorch,cuda版本比较新,我是1.6.0和10.0
2)
这个是备选,但是没用上,不太清楚。
3)这是官网,如果你是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
安装过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