tensorflow训练时指定GPU

1、在代码里面实现

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '2,3'

2、在命令行中指定

CUDA_VISIBLE_DEVICES=2 python xxx.py

3、在某一个命令窗口指定

export CUDA_VISIBLE_DEVICES="xx"

你可能感兴趣的:(tensorflow)