tensorflow中指定GPU及GPU顯存設置

查看機器上GPU情況 命令： nvidia-smi 功能：顯示機器上gpu的情況 命令： nvidia-smi -l 功能：定時更新顯示機器上gpu的情況 其中左上側有0、1、2、3的編號，表示GPU的編號，在後面指定GPU時需要使用這個編號。 在終端執行程序時指定GPU   CUDA_VISIBLE_DEVICES=1   python  your_file.py 這樣在跑你的網絡之前，告訴程序

>>阅读原文<<