2019年4月17日
摘要: 检查keras/tensorflow是否正常调用GPU代码 os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ['CUDA_VISIBLE_DEVICES'] = "0" a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a') ... 阅读全文
posted @ 2019-04-17 15:26 分享是一种快乐 阅读(3520) 评论(0) 推荐(0) 编辑