```python # limit use GPU index os.environ["CUDA_VISIBLE_DEVICES"] = "3" # '0,1' 使用 GPU 0，1 # limit GPU memory tf_config = tf.ConfigProto() # 当使用GPU时候，Tensorflow运行自动慢慢达到最大GPU的内存 tf_config.gpu_options.allow_growth = True # 当使用GPU时，设置GPU内存使用最大比例 tf_config.gpu_options.per_process_gpu_memory_fraction = 0.4 ``` 还可以参考： [link](https://www.tensorflow.org/programmers_guide/using_gpu?hl=zh-cn)