[tf] 指定运行设备
2018-12-06 本文已影响1人
VanJordan
在import tensorflow 之前进行，表示只在CPU上运行
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
设置仅在2,3,卡上运行，如果不设置的话会默认在所有GPU上执行。
os.environ['CUDA_VISIBLE_DEVICES']='2，3'
