摘要:
检查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
分享是一种快乐
阅读(3542)
评论(0)
推荐(0)
浙公网安备 33010602011771号