Deepin 15.5搭建GPU版chainer环境

论坛 期权论坛 脚本     
匿名网站用户   2020-12-19 12:59   802   0

在解决了显卡,cuda的问题后,一切都很好办了。安装chainer,安装cupy,跑训练。
需要安装的依赖,命令如下(由于使用的是大黄蜂解决方案,需要使用独显的地方,都要加optirun前缀,pip比较慢的话,可以换源,参考python换源)

pip install matplotlib
sudo apt install python-tk
optirun pip install chainer
optirun pip install cupy

然后拿官网上的例子跑一跑

wget https://github.com/chainer/chainer/archive/v3.5.0.tar.gz
tar xzf v3.5.0.tar.gz
optirun python chainer-3.5.0/examples/mnist/train_mnist.py -g 0

这里写图片描述
以我860显卡的速度,平均在190 iter/sec。如果我用cpu的话,20 iter/sec.

参考

  1. Deepin 15.5搭建GPU版caffe环境
  2. python换源
分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:1136255
帖子:227251
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP