在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'
在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'
本文标题：[tf] 指定运行设备
本文链接：https://www.haomeiwen.com/subject/dwpacqtx.html
网友评论