TensorFlow在训练模型时指定GPU进行训练_为什么我tensorflow 使用gpu0 训练,而不是gpu1

TensorflowGPU训练的时候默认选第0块GPU训练，同时占满所有卡的显存。

用tf.device()函数在指定训练时所用GPU

tf.device('/gpu:0')
1

通过CUDA_VISIBLE_DEVICES来指定(这时只用一块GPU的内存)

import os
os.environ['CUDA_VISIBLE_DEVICES']='0'
# os.environ['CUDA_VISIBLE_DEVICES']='1'
1
2
3

运行python程序前指定：

CUDA_VISIBLE_DEVICES=0 python train.py
1