2020年2月15日
TensorFlow中使用GPU
摘要： TensorFlow默认会占用设备上所有的GPU以及每个GPU的所有显存；如果指定了某块GPU,也会默认一次性占用该GPU的所有显存。可以通过以下方式解决： 1 Python代码中设置环境变量，指定GPU 本文所有代码在tensorflow 1.12.0中测试通过。 import os os.env 阅读全文
posted @ 2020-02-15 23:52 ZingpLiu 阅读(3521) 评论(0) 推荐(1) 编辑
/* 登录到博客园之后，打开博客园的后台管理，切换到“设置”选项卡，将上面的代码，粘贴到 “页脚HTML代码” 区保存即可。 */