使用tensorflow时,关于GPU的设置
查看显卡使用情况:
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))
- 上一篇 »css3与gpu加速
- 下一篇 »5个css编码坏习惯