tensorflow中指定GPU及GPU顯存設置

查看機器上GPU狀況 python 命令: nvidia-smi網絡 功能:顯示機器上gpu的狀況session 命令: nvidia-smi -l spa 功能:定時更新顯示機器上gpu的狀況blog 其中左上側有0、一、二、3的編號,表示GPU的編號,在後面指定GPU時須要使用這個編號。io 在終端執行程序時指定GPU   import CUDA_VISIBLE_DEVICES=1   pyt
相關文章
相關標籤/搜索