Tensorflow使用多个GPU时的分配方法

Tensorflow的GPU分配方法

##1.查看服务器GPU内存使用情况
输入命令行：

watch -n 10 nvidia-smi 

可以看到下面的图片显示的情况，页面每隔10s会刷新页面。

##2.Tensorflow使用多个GPU时的分配方法

方法一：在命令行中输入

CUDA_VISIBLE_DEVICES=0,1 python3 main.py

方法二：在代码中加入

import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0，1"

即可指定程序只在显卡 0,1 上运行。