个人博客转至:tybai.com

无聊就想打码,打码使我快乐


Fork me on GitHub

TensorFlow使用GPU

查看机器 GPU 的信息:

nvidia-smi

持续更新查看:

nvidia-smi -l

其他方式如下:

import os

# 使用GPU0 和 GPU1
os.environ['CUDA_VISIBLE_DEVICES'] = '0, 1'  

# 通过 allow_soft_placement 参数自动将无法放在 GPU 上的操作放回 CPU
gpuConfig = tf.ConfigProto(allow_soft_placement=True)

# 限制一个进程使用 60% 的显存
gpuConfig.gpu_options.per_process_gpu_memory_fraction = 0.6

# 运行时需要多少再给多少
gpuConfig.gpu_options.allow_growth = True  

with tf.Session(config=gpuConfig) as sess:
     pass

posted on 2018-10-09 09:35 TTyb 阅读(...) 评论(...) 编辑 收藏

导航


不用多久

我就会升职加薪

当上总经理

出任CEO

迎娶白富美

走上人生巅峰

Pulpit rock