chainer 版本差异

针对chainer4.0.0升级chainer 5.4.0后原有代码诸如chainer.cuda.to_cpu(), chainer.cuda.to_gpu(), chainer.cuda.get_device_from_id()提示找不到成员函数的问题

博主在复现一篇论文的代码时候发现其上传的code使用的chainer 4.0.0 且目测是Ubuntu系统,博主本机环境是win10 amd64系统搭配GTX1070ti卡,前面的安装cuda、cudnn等网上很多,这里不做赘述,我在安装了官网的chainer后发现其版本是5.4.0想回退到4.0.0version但是conda提示我chainercv版本不兼容,无奈硬着头皮安装了5.4.0,然后安装了cupy,这里需要说的是cupy官方解释是只支持Ubuntu和centos,但是亲测发现基于python3.6环境下win10也可以安装,需要注意的是安装cupy时候要与自己cuda version match。
安装后发现4.0chainer调用方式与5.4有些不同,对于诸如chainer.cuda.to_cpu(), chainer.cuda.to_gpu(), chainer.cuda.get_device_from_id()等函数需要改为chainer.cuda.cuda.to_cpu(), chainer.cuda.cuda.to_gpu(), chainer.cuda.cuda.get_device_from_id()追踪到底层发现chainer官方将import chainer.backends改为了from chainer.backends import cuda,所以需要多加一层cuda的调用,however,万能的python源码让我们轻松发现问题。

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值