GPU版TensorFlow设置CPU运行的办法

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
将以上三行代码放入程序首部即可。

 

 

你可能感兴趣的:(GPU版TensorFlow设置CPU运行的办法)