wgan pytorch,pyvision, py-faster-rcnn等的安装使用
因为最近在读gan的相关工作,wgan的工作不得不赞。于是直接去跑了一下wgan的代码。
原作者的wgan是在lsun上测试的,而且是基于pytorch和pyvision的,于是要装,但是由于我们一直用的是python 2.7,所以无法从WGAN的页面上跳到pytorch的官方页面下载安装,需要安装github上的版本,这个是不需要python3.5的。重新装的系统,就安装了tensorflow和caffe,编译都没问题。但是当我安装pytorch和pyvision时,涉及到lmdb,过程出了问题,conda的安装不起效果,于是折腾了一番。目前跑起来倒是很好,看了一下wgan的代码真是简单,和论文比起来简单多了。
因为之前好久没有怎么跑过代码了,听同事说有了多GPU的detection程序放出来,这个我肯定不能错过的,detection跟踪这么多年,没做什么创新性的贡献,但是至少我还是很了解行情的啊。找到代码clone下来后发现我系统新装的,cython, easydict都没有,于是继续装。这个时候就出问题了,我默认conda安装了最新版的opencv于是caffe都编译不过了,到网上找了caffe的issue发现是版本问题,于是只好重新来,装了2.4.11的版本,还有conda装了protobuf,然后我下载的程序py-R-FCN-multiGPU依然不行。最后只好跑一下我熟悉的py-faster-rcnn,可以运行,说明我的电脑环境配置没问题了。于是重新clone上面的代码,方可运行通过。
目前py-R-FCN-multiGPU 确实支持多GPU的,但是速度上并不能达到几倍的效果,具体实现原理需要再细看,目前来看是用了nvidia的NCCL库。后面我会跟着更新一些细节,看看是如何实现多GPU的。
另外,网上也看到有多GPU的版本,不过是借助于python的多process的包chainer实现的,时间比上面的版本早点,好像用的人不少,链接在这里chainer-faster-rcnn。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· winform 绘制太阳,地球,月球 运作规律
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
2016-03-10 老师喜欢的学生