2019年11月15日
tensorflow2.0 实现gpu和cpu切换
摘要： 昨天把GPU版本的tf2.0 安装成功之后，现在所有的代码运行居然都在gpu上跑了，并且在对gpu使用情况没有限制的条件下，既然gpu内存跑满了，代码就崩了怎么样才能随心所欲的指定代码是在cpu还是gpu呢首先若不加任何配置情况下，是默认使用gpu的，加上下面这句代码就使用cpu了 os.envir 阅读全文
posted @ 2019-11-15 09:54 BigZhan 阅读(8027) 评论(1) 推荐(1) 编辑