2020年1月4日
限制tensorflow训练时的显存设置，以及可见显卡号
摘要： import tensorflow as tf import os os.environ["CUDA_VISIBLE_DEVICES"] = '0' #use GPU with ID=0 config = tf.ConfigProto() config.gpu_options.per_process 阅读全文
posted @ 2020-01-04 11:58 you-wh 阅读(697) 评论(0) 推荐(0) 编辑
Fork me on GitHub