多顯卡服務器指定顯卡運行tensorflow程序

查看機器上GPU情況 命令: nvidia-smi 功能:顯示機器上gpu的情況 其中左上側有0、1、2、3的編號,表示GPU的編號,在後面指定GPU時需要使用這個編號。 方法一:在終端執行程序時指定GPU   CUDA_VISIBLE_DEVICES=1   python  your_file.py 這樣在跑你的網絡之前,告訴程序只能看到1號GPU,其他的GPU它不可見 可用的形式如下: CUD
相關文章
相關標籤/搜索