美文网首页
[tf] 指定运行设备

[tf] 指定运行设备

作者: VanJordan | 来源:发表于2018-12-06 21:48 被阅读1次

在import tensorflow 之前进行,表示只在CPU上运行

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"  
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

设置仅在2,3,卡上运行,如果不设置的话会默认在所有GPU上执行。

os.environ['CUDA_VISIBLE_DEVICES']='2,3'

相关文章

网友评论

      本文标题:[tf] 指定运行设备

      本文链接:https://www.haomeiwen.com/subject/dwpacqtx.html