会员
周边
众包
新闻
博问
闪存
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
分享是一种快乐
博客园
首页
新随笔
联系
订阅
管理
2019年4月17日
如何检查tensorflow环境是否能正常调用GPU
摘要: 检查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)
编辑