import tensorflow as tf
import os
gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
for gpu in gpus:
tf.config.experimental.set_memory_growth(gpu, True)
os.environ['CUDA_VISIBLE_DEVICES']='2'
tf2.0 指定GPU,并设置显存分配
最新推荐文章于 2022-07-25 23:36:45 发布