0
点赞
收藏
分享

微信扫一扫

在CPU上运行Tensorflow

佳简诚锄 2022-07-18 阅读 57

如果你是用的GPU版本的Tensorflow,你可以这样来使用CPU版本的Tensorlfow:

config = tf.ConfigProto(
device_count = {'GPU': 0}
)
sess = tf.Session(config=config)

 

黄世宇/Shiyu Huang's Personal Page:​​https://huangshiyu13.github.io/​​



举报

相关推荐

0 条评论