(原)caffe在ubuntu中设置GPU的ID号及使用多个GPU
Ubuntu 设置 多个 ID GPU Caffe 使用
2023-09-11 14:22:24 时间
转载请注明出处:
http://www.cnblogs.com/darkknightzh/p/5884892.html
参考网址:
http://caffe.berkeleyvision.org/tutorial/interfaces.html
由于在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越多,开始初始化时时间越久。当然,训练速度越快。
相关文章
- Ubuntu设置开机启动
- 在 Ubuntu 20.04 上设置默认 Java 版本
- 设置 Ubuntu 系统自动关机小工具 EasyShutdown
- 解决Ubuntu 14.04 Unity桌面环境登录后冻结问题
- Ubuntu Touch 和 Android 双系统安装官方指南
- 在Ubuntu 14.04 上使用Bacula和Webmin设置备份服务器
- [快速技巧]通过命令在 Debian/Ubuntu 中设置默认浏览器
- 如何在Ubuntu 14.04 上为Apache 2.4 安装SSL支持
- 如何解决 Ubuntu 下不能记住亮度设置的问题
- 在 Ubuntu 15.10 上为单个网卡设置多个 IP 地址
- Ubuntu 14.04 安装jdk,tomcat
- 在docker build过程中执行apt-get update时报错"Could not connect to archive.ubuntu.com:80"如何处理?
- ubuntu 18.04 64bit下如何安装python开发工具jupyter
- VNC连接远程Ubuntu设置
- ubuntu 16.04 更改默认Python版本
- Ubuntu双网卡设置内外网上网的问题
- Ubuntu-1604-LTS在虚拟机设置分辨率
- Ubuntu 18.04 启用 rc.local 设置开机启动
- android在ubuntu中编译为.apk资料
- Ubuntu 22.04 开启SSH
- 如何在Ubuntu系统中进行系统级的代理设置
- 如何修复ubuntu的uefi启动——如何将Ubuntu安装入移动硬盘中
- Ubuntu Server无桌面无显示器情况下虚拟屏幕xvfb的安装及设置—ubuntu18.04server服务器系统下为python安装虚拟显示器 (使用jupyter notebook在web端播放openai的gym下保存的运行视频——需安装ipython)
- [转载] Ubuntu上Firefox字体太小--高分屏背锅——高清分辨率屏幕下浏览器设置
- Ubuntu 16.04 LTS安装 cuda8.0
- ubuntu 环境变量设置
- OpenCV 2.4.13 installed in Ubuntu 14 and CMakeLists Demo
- 3. Ubuntu系统中文化、安装中文输入法、解决VNC上ubuntu后无法复制粘贴问题
- tomcat在ubuntu中设置开机启动
- 【职场技能】Ubuntu设置屏幕睡眠系统不休眠
- Ubuntu 16.04 设置MySQL远程访问权限