Tensorflow
内推小仙女
字节跳动靠谱内推达人~ 校招实习社招,字节全岗位都可内推哦~ 欢迎找我内推!
展开
-
tensorflow ConfigProto&GPU
tensorflow ConfigPrototf.ConfigProto一般用在创建session的时候。用来对session进行参数配置with tf.Session(config = tf.ConfigProto(...),...)1#tf.ConfigProto()的参数log_device_placement=True : 是否打印设...转载 2018-06-20 20:05:53 · 376 阅读 · 0 评论 -
Tensorflow 多GPU下,指定使用某一块GPU, 多块, 禁用
如果机器中有多块GPU,tensorflow会默认吃掉所有能用的显存, 如果实验室多人公用一台服务器,希望指定使用特定某块GPU。可以在文件开头加入如下代码:import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"os.enviro...转载 2018-06-18 20:37:34 · 1174 阅读 · 0 评论