如何在服务器上使用GPU跑代码（tensorflow相关）

在代码加入这一行：

os.environ['CUDA_VISIBLE_DEVICES'] = '0'
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'

查看使用的是GPU还是CPU：

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

查看GPU使用率：nvidia-smi