import os
import tensorflow as tf
os.environ['CUDA_VISIBLE_DEVICES']='0' #指定GPU
config = tf.ConfigProto()
config.gpu_option.allow_growth = True #开启增量式使用(用多少拿多少)
seesion = tf.Seesion(config=config,...)
Tensorflow多人共享服务器GPU的小技巧
最新推荐文章于 2024-09-24 10:27:55 发布