TensorFlow中使用GPU

TensorFlow默认会占用设备上所有的GPU以及每个GPU的所有显存；如果指定了某块GPU,也会默认一次性占用该GPU的所有显存。可以通过以下方式解决：

1 Python代码中设置环境变量，指定GPU

import os os.environ["CUDA_VISIBLE_DEVICES"] = "2" # 指定只是用第三块GPU