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
Hadoop|
Apache Pig|
Apache Kafka|
Apache Storm|
Impala|
Zookeeper|
SAS|
TensorFlow|
人工智能基础|
Apache Kylin|
Openstack|
Flink|
MapReduce|
大数据|
云计算|
用户登录
还没有账号?立即注册
用户注册
投稿取消
文章分类: |
|
还能输入300字
上传中....