如何指定GPU运行程序
2017-12-07 10:38
267 查看
连服务器有时候会出现out of memory解决方法:换用空闲的gpu1. 查看GPUnvidia-smi -L2. 查看7号GPUnvidia-smi -q -i 7如果只看memory情况。可以用:nvidia-smi -q -i 7 -d MEMORY3.设置GPUidCUDA_VISIBLE_DEVICES=6(或CUDA_VISIBLE_DEVICES=6,7)command举例说明:版权归作者所有,任何形式转载请联系作者。作者:宝图(来自豆瓣)来源:https://www.douban.com/note/641416299/
这条命令适用于命令行运行tensorflow程序的时候,指定gpu.
只需要在命令之前设置环境变量,简单来说比如原本程序是命令行运行pythontrain.py
假定这里gpu总共有八块,通过nvidia-smi查看发现5,6,7是空闲的(从0开始编号)
则运行命令修改为:CUDA_VISIBLE_DEVICES=5,6,7 python train.py
这条命令适用于命令行运行tensorflow程序的时候,指定gpu.
只需要在命令之前设置环境变量,简单来说比如原本程序是命令行运行pythontrain.py
假定这里gpu总共有八块,通过nvidia-smi查看发现5,6,7是空闲的(从0开始编号)
则运行命令修改为:CUDA_VISIBLE_DEVICES=5,6,7 python train.py
相关文章推荐
- 如何指定程序在某个GPU上运行
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- [python] 在ubuntu中, 如何运行指定位置的py程序
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 如何在u盘插上以后就自动运行指定的程序?
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 转载:如何指定程序在Vista上面需要提升权限运行(Elevated)
- 如何注册程序在指定时间运行