使用gpu跑代码

1. 配置tensorflow-gpu环境
2. 代码里选择指定gpu【可选】

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '0'      # 使用 GPU 0
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'    # 使用 GPU 0，1

1. 终端设置可见gpu【可选】

CUDA_VISIBLE_DEVICES=1   python  your_file.py
# 可用的形式如下：
# CUDA_VISIBLE_DEVICES=1           Only device 1 will be seen
# CUDA_VISIBLE_DEVICES=0,1         Devices 0 and 1 will be visible
# CUDA_VISIBLE_DEVICES="0,1"       Same as above, quotation marks are optional
# CUDA_VISIBLE_DEVICES=0,2,3       Devices 0, 2, 3 will be visible; device 1 is masked

# CUDA_VISIBLE_DEVICES=""          No GPU will be visible

1. 查看机器上GPU情况

nvidia-smi
# 显示机器上gpu的情况
nvidia-smi -l
# 定时更新显示机器上gpu的情况

【参考】
tensorflow指定GPU运算
Tensorflow指定GPU的指定和显存设置