你好啊渔夫

  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

查看显卡使用情况:

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)) 

posted on 2019-06-06 13:21  你好啊渔夫  阅读(1024)  评论(0编辑  收藏  举报