|
在解决了显卡,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.
参考
- Deepin 15.5搭建GPU版caffe环境
- python换源
|