tensorflow指定gpu进行训练

服务器共用的时候需要进行gpu分配,每个人用哪几块。

方法1

在训练代码内加上:

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

方法2

在训练指令前面加上CUDA_VISIBLE_DEVICES=1,即:

CUDA_VISIBLE_DEVICES=1 python train.py
……

HOPE U ENJOY

你可能感兴趣的:(tensorflow)