您的位置:首页 > 其它

如何指定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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: