tensorflow tf.device
- 作者: 周猩猩
- 来源: 51数据库
- 2020-10-05
with tf.Session() as sess:
with tf.device("/cpu:0"):
后面写你的代码
如果是新手的话,建议安装tensorflow而不是tensorflow-gpu
希望可以帮到你
这很简单,tensorflow-gpu是使用CUDA的
设置CUDA参数为-1就行,-1就是不使用GPU
代码的最後一行就是了,建议全部复制贴上
import?os
os.environ['TF_CPP_MIN_LOG_LEVEL']='3'?#Disable?Tensorflow?debugging?information
os.environ["CUDA_DEVICE_ORDER"]?=?"PCI_BUS_ID"
os.environ['CUDA_VISIBLE_DEVICES']?=?'-1'?#Set?GPU?device?-1
with tf.device("/cpu:0"):
后面写你的代码
如果是新手的话,建议安装tensorflow而不是tensorflow-gpu
希望可以帮到你
这很简单,tensorflow-gpu是使用CUDA的
设置CUDA参数为-1就行,-1就是不使用GPU
代码的最後一行就是了,建议全部复制贴上
import?os
os.environ['TF_CPP_MIN_LOG_LEVEL']='3'?#Disable?Tensorflow?debugging?information
os.environ["CUDA_DEVICE_ORDER"]?=?"PCI_BUS_ID"
os.environ['CUDA_VISIBLE_DEVICES']?=?'-1'?#Set?GPU?device?-1
推荐阅读