分类：全局设置

2020
10-08

Tensorflow全局设置可见GPU编号操作

笔者需要tensorflow仅运行在一个GPU上（机器本身有多GPU），而且需要依据系统参数动态调节，故无法简单使用CUDA_VISIBLE_DEVICES。一种方式是全局使用tf.device函数生成的域，但设备号需要在绘制Graph前指定，仍然不够灵活。查阅文档发现config的GPUOptions中的visible_device_list可以定义GPU编号从visible到virtual的映射，即可以设置tensorflow可见的GPUdevice，从而全局设置了tensorflow可见的GPU编号。代码如下：config=tf... 继续阅读 >