使用tensorflow时，关于GPU的设置

标签：_id   str   options   port   conf   cuda   imp   使用   smi   

查看显卡使用情况：

nvidia-smi

设置tensorflow按需分配资源：

import os

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"

os.environ["CUDA_VISIBLE_DEVICES"] = "1" # 使用第二块GPU（从0开始）

gpu_options = tf.GPUOptions(allow_growth=True)

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) 

使用tensorflow时，关于GPU的设置

标签：_id   str   options   port   conf   cuda   imp   使用   smi   

原文地址：https://www.cnblogs.com/hellojiaojiao/p/10984422.html