2019年4月24日
【TensorFlow】使用指定的GPU
摘要： 现有的模型训练中，一般config文件会指定需要使用的GPU。 但是在某些程序中，可能代码写的有些问题，即使指定了GPU，也会存在使用了所有空闲GPU的情况。 其实如果机器中有多块GPU，那么TensorFlow默认会占满所有能用的显存，而在需要运行多个项目时，我们希望可以使用特定的某块GPU。 下 阅读全文
posted @ 2019-04-24 17:55 EddiePing 阅读(541) 评论(0) 推荐(0)