最新消息:雨落星辰是一个专注网站SEO优化、网站SEO诊断、搜索引擎研究、网络营销推广、网站策划运营及站长类的自媒体原创博客

pwc

网站源码admin34浏览0评论

pwc

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
发布评论

评论列表(0)

  1. 暂无评论