首页 » 2017 » 十二月

Monthly: 十二月 2017

Python下在多显卡机器上指定Tensorflow使用哪几块设备

vForce | Keras, Python | 2017-12-12
以下代码指定使用0号和1号显卡,分配70%的显存。实际使用时请根据显卡的互相连接情况调整设备号 import os import tensorflow as tf from keras.backend.tensorflow_backend import set_session os.environ["CUDA_VISIBLE_DEVICES"] = '0, 1' config = tf.ConfigProto() config.gpu_options.allocator_type = 'BFC' #A &q... [阅读全文]
ė 6没有评论 0
Ɣ回顶部