tensorflow gpu训练objectdetectionapi 卡住不动了 gpu占满利用率0是什么原因
不知道你这个问题是否已经解决, 如果还没有解决的话:这条命令适用于命令行运行tensorflow程序的时候,指定gpu.
只需要在命令之前设置环境变量,简单来说比如原本程序是命令行运行pythontrain.py
假定这里gpu总共有八块,通过nvidia-smi查看发现5,6,7是空闲的(从0开始编号)
则运行命令修改为:CUDA_VISIBLE_DEVICES=5,6,7 python train.py