lunix下跑caffe用到的一些命令

最新推荐文章于 2018-10-19 22:42:51 发布

原创最新推荐文章于 2018-10-19 22:42:51 发布 · 389 阅读

1 ·

CC 4.0 BY-SA版权

深度学习入门专栏收录该内容

12 篇文章

订阅专栏

本文介绍如何使用Caffe工具包中的plot_training_log.py脚本来绘制训练过程中的各种曲线图，包括准确率、损失及学习率等，并提供nvidia-smi命令查看GPU内存使用情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

显示所用内存：nvidia-smi (读一次）
nvidia-smi -lms(每秒读一次)

显示准确率曲线：python ../../../tool/extra/plot_training_log.py.example 0 0.png 1.log

显示准loss曲线：python ../../../tool/extra/plot_training_log.py.example 6 0.png 1.log

利用extra里面的一个plot_training_log.py.example文件画出图。
这个文件的使用格式是：
./caffe/tools/extra/plot_training_log.py.example “0—7” “输出照片的名字”　caffe.usrname-Ubuntu.usrname.log.INFO.当前日期-当前时间.xxxx.log
第一个参数是0—7的数字，每个数字代表绘制一种曲线图。
第二个参数就是要输出图片的名字。
第三个参数就是我们之前利用GLOG_logtostderr=0得到的文件，只是需要在这个文件的后面再加上个.log，如果不加上这个.log的话会出现报错，画出图片的标题将是.log之前的文字，所以大家可以根据自己想要的title更改一下文件的名字，并在程序中相应得修改即可。

其中0—7数字所代表的画图的种类为：
Supported chart types:
0: Test accuracy vs. Iters
1: Test accuracy vs. Seconds
2: Test loss vs. Iters
3: Test loss vs. Seconds
4: Train learning rate vs. Iters
5: Train learning rate vs. Seconds
6: Train loss vs. Iters
7: Train loss vs. Seconds

执行.py文件：在终端中输入命令：python **.py 即可

出现no module named caffe或者是no module named caffe-proto，则在终端输入：

export PYTHONPATH=$PYTHONPATH:=/home/**（您服务器的名字）/caffe/python