Tensorflow指定GPU的指定和顯存設置

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

>>阅读原文<<