前端之家收集整理的这篇文章主要介绍了
Ubuntu下Caffe安装记录,
前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
总体过程
命令行训练Mnist
./data/mnist/get_mnist.sh
sudo ./examples/mnist/create_mnist.sh
sudo ./examples/mnist/train_lenet.sh
PyCharm训练Mnist
- 需要用sudo执行pycharm,还是防止出现Cannot create Cublas handle. Cublas won’t be available.错误;
- 需要将example/mnist/下的一些文件中定义的训练数据等路径配置改成绝对路径;
代码:
import caffe
caffe.set_mode_gpu()
solver = caffe.SGDSolver('/home/guopin/caffe/examples/mnist/lenet_solver.prototxt')
solver.solve()
原文链接:https://www.f2er.com/ubuntu/350088.html