如何设置Tensorflow仅使用特定编号的GPU？

本文共计834个文字，预计阅读时间需要4分钟。

作者需确保TensorFlow仅在单个GPU上运行（机器本身拥有多个GPU），且需依赖系统参数动态调整。直接使用CUDA_VISIBLE_DEVICES方法较为复杂。一种简便方法是全局使用tf.device函数创建区域，但设备编号需在引号内指定。

笔者需要tensorflow仅运行在一个GPU上（机器本身有多GPU），而且需要依据系统参数动态调节，故无法简单使用CUDA_VISIBLE_DEVICES。

一种方式是全局使用tf.device函数生成的域，但设备号需要在绘制Graph前指定，仍然不够灵活。

查阅文档发现config的GPUOptions中的visible_device_list可以定义GPU编号从visible到virtual的映射，即可以设置tensorflow可见的GPU device，从而全局设置了tensorflow可见的GPU编号。