caffe,gpu ID号使用方法
来源:互联网 发布:市场营销大数据分析 编辑:程序博客网 时间:2024/06/10 08:32
由于在ubuntu中使用caffe的程序时,都使用.sh文件,该文件中常见的命令为:
./build/tools/caffe train --solver=examples/testXXX/solver.prototxt
当电脑中有多个GPU时,默认使用GPU0。如果想使用其他的GPU,可以将该文件内容修改如下:
./build/tools/caffe train --solver=examples/testXXX/solver.prototxt --gpu 2
注意,caffe中默认编号从0开始,因而--GPU 2的意思是使用第3个GPU。
如果要使用多个GPU,可以使用如下命令:
./build/tools/caffe train --solver=examples/testXXX/solver.prototxt --gpu 0,1,2,3
则使用0,1,2,3这4个GPU。
如果要使用所有的GPU,可使用如下命令:
./build/tools/caffe train --solver=examples/testXXX/solver.prototxt --gpu all
注意,使用的GPU越多,开始初始化时时间越久。当然,训练速度越快。
0 0
- caffe,gpu ID号使用方法
- Caffe::SetDevice 设置GPU的ID号
- caffe中多GPU的使用方法
- 在使用caffe的时候,如何设置GPU的ID号
- 【Caffe】ubuntu14.04+caffe+gpu
- opensuse caffe GPU 配置
- ubuntu14.04+GPU+caffe
- caffe 去掉GPU依赖
- caffe 指定GPU
- caffe训练GPU配置
- caffe多GPU上手
- caffe --Multi-GPU Usage
- ubuntu16 caffe GPU
- ubuntu安装caffe +gpu
- windows caffe gpu 配置
- caffe配置 windows gpu
- caffe ssd gpu
- caffe训练时总是会占用0号GPU卡
- java与C++的内联函数
- Ubuntu16.04完全卸载Mysql 5.7
- Numpy与统计学
- UE4中实现描边效果
- jQuery筛选--查找,串联
- caffe,gpu ID号使用方法
- 替换jqgrid 分页小箭头样式
- 第十六周 职工信息结构体
- 快速排序【JAVA实现】
- 数据库范式
- Flutter基础—定位对齐之大小比例
- NYOJ311完全背包
- 18. 4Sum
- (转)HTTP协议—— 简单认识TCP/IP协议