- 配置tensorflow-gpu环境
- 代码里选择指定gpu【可选】
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '0' # 使用 GPU 0
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' # 使用 GPU 0,1
- 终端设置可见gpu【可选】
CUDA_VISIBLE_DEVICES=1 python your_fil
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '0' # 使用 GPU 0
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' # 使用 GPU 0,1
CUDA_VISIBLE_DEVICES=1 python your_fil