Linux查看GPU及tensorflow指定GPU

查看GPU 命令: nvidia-smi 功能:顯示機器上gpu的情況 命令: nvidia-smi -l 功能:定時更新顯示機器上gpu的情況 指定GPU import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "0" # 使用第一塊GPU(從0開始)
相關文章
相關標籤/搜索